Python/Matplotlib - 颜色条配置

Question

这个问题已经有答案了:

matplotlib colorbar for scatter (3 个回答)

已关闭 5 年前。

我有一个关于 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()