自定义修改 plt.scatter 中 c=ground_truth_support 的默认颜色
要自定义修改plt.scatter中c=ground_truth_support的默认颜色,可以通过传递一个颜色映射对象给参数cmap来实现。
首先,你需要创建一个颜色映射对象。可以使用matplotlib.colors.ListedColormap来创建一个自定义颜色映射。该函数接受一个颜色列表作为参数,每个元素表示一个颜色。
然后,你需要将cmap参数传递给plt.scatter函数,以指定使用自定义的颜色映射。
下面是一个示例代码:
import matplotlib.pyplot as plt
import matplotlib.colors as mcolors
import numpy as np
# 创建自定义颜色映射
colors = ['red', 'green', 'blue'] # 自定义颜色列表
cmap = mcolors.ListedColormap(colors)
# 创建示例数据
sample_tsne = np.random.rand(100, 2)
ground_truth_support = np.random.randint(0, 3, 100)
# 绘制散点图
plt.scatter(sample_tsne[:, 0], sample_tsne[:, 1], c=ground_truth_support, cmap=cmap, marker='s', s=200)
# 显示图形
plt.show()
在上述示例中,我们创建了一个包含三种颜色的颜色列表,并使用mcolors.ListedColormap函数创建了一个颜色映射对象cmap。然后,我们将该cmap对象传递给plt.scatter函数的cmap参数,以指定使用自定义的颜色映射。
你可以根据自己的需求修改颜色列表colors,以及其他plt.scatter函数的参数来自定义修改散点图的样式。
原文地址: https://www.cveoy.top/t/topic/fOrQ 著作权归作者所有。请勿转载和采集!