matplotlib 双色散点图[重复]

cunj1qz1 于 2023-05-18 发布在其他

关注(0)|答案(1)|浏览(148)

此问题已在此处有答案：

Setting different color for each series in scatter plot on matplotlib（8个回答）
4年前关闭。
我需要给数组的一半涂上一种颜色。
假设我有两个数组，我连接

arr1 = [[1,2,
         3,4]]
arr2 = [[1,2,
         3,4]]

arr 3 =  np.concatenate((arr1, arr2), axis=0)

plt.scatter(arr3[:,0], arr3[:,1], c= ???)

这是一个例子，我的真实的数据集有16，000个数据点，因此单独的标签数组可能是不可能的。从指数0 - 8000~我想变红。8000 - 16000我想变蓝。谢谢

1条答案

一个简单的例子来说明如何做到这一点，假设你有一个由16，000个元素组成的数组和两个维度。可以在同一图形中分别绘制两组点，如下所示：

plt.scatter(arr3[0:8000, 0], arr3[0:8000, 1], c="red")
plt.scatter(arr3[8000:16000, 0],arr3[8000:16000, 1], c="blue")

plt.show()

这两组点将以不同的颜色显示在同一图形中。您只需要修改索引，使其适合您的数据。