如何解决如何使用数组为散点图中的群集着色?
我正在使用sklearn
noisy_circles
function(如链接中的1500点)来创建2个圆圈。
我有一个点引用数组,其中每个值都应该是不同的簇颜色:
nbrs_array = [ 88,88,973,984,992,972,88]
数组中有5个唯一值,因此应该有5种不同的颜色。但是当我使用以下方式绘制它时:
plt.figure(figsize=(10,10))
plt.scatter(x,y,c=nbrs_array)
plt.show()
输出为
解决方法
类似的事情应该可以完成:
import numpy as np
from matplotlib import cm
from matplotlib import pyplot as plt
from matplotlib.colors import Normalize,to_hex
from sklearn import datasets
def get_colors(arr,cmap='viridis'):
cmap = cm.get_cmap(cmap)
n = len(np.unique(arr))
colornorm = Normalize(vmin=1,vmax=n)
hex_map = dict()
for i,cl in enumerate(np.unique(arr)):
hex_map[cl] = to_hex(cmap(colornorm(i + 1)))
colors = list(map(lambda x: hex_map[x],arr))
return colors
n_samples = 1500
data,_ = datasets.make_circles(n_samples=n_samples,factor=.5,noise=.05)
x,y = data[:,0],data[:,1]
nbrs_array = [...]
plt.figure(figsize=(10,10))
plt.scatter(x,y,c=get_colors(nbrs_array))
plt.show()
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。