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

时间:2017-02-26 13:27:38

标签: python matplotlib

我在下面运行此代码后。

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()

上面的代码显示如下图像: enter image description here

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

1 个答案:

答案 0 :(得分:2)

当然,变量cm并未在脚本中的任何位置使用,因此您可以在不更改任何内容的情况下对其进行注释。

cm = plt.cm.RdBu表示色彩映射,就像后面定义的cm_bright = ListedColormap(..)一样。所以你可能想要改用它:

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

有关色彩映射的完整列表,请参阅matplotlib colormap reference