cm = plt.cm.RdBu是什么意思？

Question

在我运行下面的代码之后。

import matplotlib.pyplot as plt
from matplotlib.colors import ListedColormap
cm = plt.cm.RdBu
cm_bright = ListedColormap(['#FF0000', '#0000FF'])
ax = plt.subplot()  

ax.set_title("Input data")
# Plot the training points
ax.scatter(X[:, 0], X[:, 1], c=y, cmap=cm_bright)
ax.set_xticks(())
ax.set_yticks(())
plt.tight_layout()
plt.show()

上面的代码显示的图像如下：

​

但是，当我注释代码cm = plt.cm.RdBu时，结果不会改变。所以，问题是代码cm = plt.cm.RdBu的用法是什么？

Answer 1

当然，变量cm不在脚本中使用，所以您可以在不更改任何内容的情况下将其注释掉。

cm = plt.cm.RdBu表示一个彩色映射，就像后面定义的cm_bright = ListedColormap(..)一样。因此，您可能需要使用它来代替：

ax.scatter(..., cmap=cm)

有关颜色图的完整列表，请参见matplotlib颜色图引用。