python - Matplotlib 散点图；颜色作为第三个变量的函数

Question

我想制作一个散点图(使用 matplotlib)，其中的点根据第三个变量进行着色。我已经非常接近了:

plt.scatter(w, M, c=p, marker='s')

其中 w 和 M 是数据点，p 是我要对其进行着色的变量。
但是我想用灰度而不是彩色来做。有人可以帮忙吗？

Answer 1

无需手动设置颜色。相反，请指定灰度颜色图...

import numpy as np
import matplotlib.pyplot as plt

# Generate data...
x = np.random.random(10)
y = np.random.random(10)

# Plot...
plt.scatter(x, y, c=y, s=500)
plt.gray()

plt.show()

或者，如果您更喜欢 wider range of colormaps ，您还可以将 cmap kwarg 指定为 scatter。要使用其中任何一个的反转版本，只需指定其中任何一个的“_r”版本。例如。 gray_r 而不是 gray。预先制作了几种不同的灰度颜色图(例如 gray、gist_yarg、binary 等)。

import matplotlib.pyplot as plt
import numpy as np

# Generate data...
x = np.random.random(10)
y = np.random.random(10)

plt.scatter(x, y, c=y, s=500, cmap='gray')
plt.show()