Python matplotlib : Add legend for a colormap

Question

我是 matplotlib 的新手，我正在尝试重新创建一个如下图所示的对数对数图:

我知道我必须使用颜色图根据我的 x 和 y 值创建这样的输出。我想不通的是如何添加图例，以便它可以动态计算颜色图变量的值(例如，此处称为“Degree”)。到目前为止，这是我得到的示例图:

from matplotlib import pyplot as plt

x = [1, 2, 3, 4, 5, 6, 7, 8, 9]
y = [125, 32, 54, 253, 67, 87, 233, 56, 67]

color = [str(item/255.) for item in y]

plt.scatter(x, y, s=500, c=color)

plt.show()

任何帮助将不胜感激。

Answer 1

您可以为绘图分配适当的颜色图，在本例中为灰度颜色图:

x = [1, 2, 3, 4, 5, 6, 7, 8, 9]
y = [125, 32, 54, 253, 67, 87, 233, 56, 67]

color = [item / 255.0 for item in y]
grayscale_map = plt.get_cmap('gray')

plt.figure()
plt.scatter(x, y, s = 500, c = color, cmap = grayscale_map)
plt.colorbar()
plt.show()

请注意，我将您的 color 列表更改为(大概)范围从 0 到 1 的 float 列表。