一种从数组中提取数据的pythonic方法

Question

是否有更多pythonic方法从此txt文件中提取数据？ 事先声明全局变量似乎很麻烦，并且使用范围迭代而不是使用python进行i方法迭代。

data1 = np.loadtxt("testProfil5.txt",float,delimiter=None)
x,y = [],[]
for i in range(np.size(data1)/2):
    x.append(data1[i][0])
    y.append(data1[i][1])

plt.plot(x,y)

虽然这有效，但我会欣赏更多的pythonic方法。我知道有大熊猫和numpy阵列......

Answer 1

如果有包含数据的NumPy数组，请尽可能避免使用Python列表。在你的情况下：

x = data1[:,0]
y = data1[:,1]

然后您可以直接绘制数据，无需复制。

P.S。：如果有一天你确实需要np.size(data1)/2，你可以简单地说len(data1)，因为你所做的只是得到行数（第一维）。