Python/Matplotlib - Colorbar配置

Question

我有一个关于matplotlib的颜色条的问题。

散点设置为x、y和大小。我的意图是绘制200个随机大小和随机颜色的点。颜色是用200*3矩阵给出的。图片是正确的，而颜色栏显示错误。

我意识到colorbar()输入类型有问题。但我不知道如何配置或修复它。有谁能帮帮我吗？谢谢。

import numpy as np
import matplotlib.pyplot as plt

x = np.random.rand(200)
y = np.random.rand(200)
size = np.random.rand(200) * 200

color = np.array(np.random.rand(200*3))
color.shape = 200, 3
pic = plt.scatter(x, y, size, color)
plt.colorbar()

Answer 1

颜色条表示将某些值映射到颜色。

在问题的代码中，你没有任何值，只有颜色。所以matplotlib当然不知道要在颜色栏中映射什么值。

相反，您可以指定要映射的值。然后matplotlib将使用默认的颜色映射(或者您可以使用cmap参数指定的颜色映射)。将这些值映射到色彩映射表中的尊重颜色。

import numpy as np
import matplotlib.pyplot as plt

x = np.random.rand(200)
y = np.random.rand(200)
size = np.random.rand(200) * 200

value = np.random.rand(200)

pic = plt.scatter(x, y, size, value)
plt.colorbar(pic)

plt.show()

​

​