Matplotlib 更改图形颜色

Question

我想在下一个图表中更改颜色。我需要黄色为绿色，绿色为红色，紫色为蓝色。

我有下一个代码：

fig = plt.figure()
ax = fig.add_subplot(111)

scatter = ax.scatter( kmeans['Revenue'], kmeans['Frequency'],c=kmeans['Cluster'], s=15)
                
ax.set_xlabel('Recency')
ax.set_ylabel('Frequency')
plt.colorbar(scatter)

如何更改颜色？

Answer 1

如果您要阅读 scatter 的文档，那么您会看到 cmap 的选项 colormap。

如果您使用 Google matplotlib scatter colormap 那么您会发现 Choosing Colormaps in Matplotlib

它显示了预定义的颜色图，看来您需要 brg。

---

import pandas as pd
import numpy as np
import matplotlib.pyplot as plt

random = np.random.sample((100,3))
kmeans = pd.DataFrame(random, columns=['Revenue', 'Frequency', 'Cluster'])

fig = plt.figure()
ax = fig.add_subplot(111)

scatter = ax.scatter( kmeans['Revenue'], kmeans['Frequency'], c=kmeans['Cluster'], s=15, cmap='brg')
                
ax.set_xlabel('Recency')
ax.set_ylabel('Frequency')
plt.colorbar(scatter)

plt.show()