Matplotlib散点图有两种颜色

Question

我需要为数组的一半上色一种颜色。

假设我有2个串联的数组

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我想成为蓝色。谢谢

Answer 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()

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