如何在python中将ndarray转换为整数

Question

我有一个代码，其中有5个ndarray，需要在matplotlib.上对其进行可视化处理，首先，我仅尝试绘制一个ndarray，如下所示：

x = data['embeddings'][0]
y = np.linspace(0, 50, 128)
plt.scatter(x, y)
plt.show()

看起来像下面的东西：

现在的问题是，如果我还绘制其他4个ndarray，此图将看起来非常混乱，并且我将无法检查哪个ndarray是哪个。所以我想知道是否有任何方法可以将ndarray转换或映射为int，以便当我们显示它时，在图形上只为一个ndarray绘制了一个点。请提出一些好的解决方案。谢谢

Answer 1

如果需要区分为所有nd数组绘制的数据，则可以将标签和图例一起使用。

x = data['embeddings'][0]
y = np.linspace(0, 50, 128)
plt.scatter(x, y, label="FirstGraph")

x2 = ....
y2= ....
plt.scatter(x2, y2, label="SecondGraph")
plt.legend() 
plt.show()

通过这种方式，您将能够区分为不同的nd数组绘制的数据。