英文:
Unpack matrices in numpy file and plot them
问题
I need a code that reads a .npy file containing a set of arrays and creates a scatter plot using numpy and matplotlib. The first column need to be the x cordinates and the second column are the y cordinates.
import numpy as npimport matplotlib.pyplot as plt# Load the .npy filea = np.load("filename.npy")# Extract x and y coordinatesx = a[:, :, 0]y = a[:, :, 1]# Create a scatter plotplt.scatter(x, y)plt.xlabel('X Coordinates')plt.ylabel('Y Coordinates')plt.show()
英文:
I need a code that reads a .npy file containing a set of arrays and creates a scatter plot using numpy and matplotlib. The first column need to be the x cordinates and the second column are the y cordinates
ex:
a = np.load("filename.npy")
a([[[19.3, 117.88],[20.77, 118.99],[18.92, 117.66],[16.23, 115.67]],[[16.335, 113.789],[17.876, 116.8],[22.76, 115.34],[23.33, 111.45]],[[22.56, 113.76],[21.6, 118.07],[18.3, 116.60],[21.739, 117.903]]])
答案1
得分: 1
只需切片“columns”（实际上是第三维度），然后将其传递给plt.scatter。Matplotlib会在内部展平输入。
import matplotlib.pyplot as pltplt.scatter(a[:, :, 0], a[:, :, 1])
巧妙的替代方法：
plt.scatter(*a.reshape(-1, 2).T)
输出：
英文:
Simply slice the "columns" (actually the third dimension) and pass it to plt.scatter. Matplotlib will flatten the inputs internally.
import matplotlib.pyplot as pltplt.scatter(a[:, :, 0], a[:, :, 1])
Hacky alternative:
plt.scatter(*a.reshape(-1, 2).T)
Output:
通过集体智慧和协作来改善编程学习和解决问题的方式。致力于成为全球开发者共同参与的知识库，让每个人都能够通过互相帮助和分享经验来进步。
评论