我在下面运行此代码后。
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
的用法是什么?
答案 0 :(得分:2)
当然,变量cm
并未在脚本中的任何位置使用,因此您可以在不更改任何内容的情况下对其进行注释。
cm = plt.cm.RdBu
表示色彩映射,就像后面定义的cm_bright = ListedColormap(..)
一样。所以你可能想要改用它:
ax.scatter(..., cmap=cm)
有关色彩映射的完整列表,请参阅matplotlib colormap reference。