修改 plt.scatter() 默认颜色:使用颜色映射
要修改 plt.scatter 函数的默认颜色,可以使用参数 cmap 来指定一个颜色映射。颜色映射是一个从数据值到颜色的映射关系。首先,你需要创建一个颜色映射对象,然后将它作为参数传递给 cmap。下面是一个示例:
import matplotlib.pyplot as plt
import numpy as np
# 创建一个颜色映射对象
cmap = plt.cm.get_cmap('cool')
# 生成一些样本数据
sample_tsne = np.random.rand(100, 2)
ground_truth_support = np.random.randint(0, 10, 100)
# 使用scatter函数绘制散点图,并指定颜色映射
plt.scatter(sample_tsne[:, 0], sample_tsne[:, 1], c=ground_truth_support, marker='s', s=200, cmap=cmap)
# 添加颜色条
plt.colorbar()
# 显示图形
plt.show()
在上面的示例中,我们使用 plt.cm.get_cmap('cool') 创建了一个颜色映射对象,并将其命名为 cmap。然后,我们将 cmap 作为参数传递给了 scatter 函数的 cmap 参数。这将使散点图的颜色根据 ground_truth_support 的值从颜色映射中选择对应的颜色。最后,我们使用 plt.colorbar() 函数添加了一个颜色条,以便查看颜色与值之间的对应关系。
您可以根据自己的需求选择不同的颜色映射,例如 'cool'、'hot'、'jet' 等。您还可以使用 plt.cm.get_cmap() 函数来创建自定义的颜色映射。
原文地址: https://www.cveoy.top/t/topic/fOrG 著作权归作者所有。请勿转载和采集!