使用numpy读取两列的csv文件

Question

我很好奇是否有更好的方法来做到这一点。我有一个包含两列的 csv 文件（这相当常见），例如第一列是时间戳，第二列是数据。

# temp.csv
0., 0.
1., 5.
2., 10.
3., 15.
4., 10.
5., 0.

然后我想读取这个 temp.csv 文件：

import numpy as np

my_csv = np.genfromtxt('./temp.csv', delimiter=',')
time = my_csv[:, 0]
data = my_csv[:, 1]

这完全没问题，但我只是好奇是否有更优雅的方法来做到这一点，因为这是一种相当常见的做法。

谢谢你！

-肖恩

Answer 1

你可以做：

my_csv = np.genfromtxt('./temp.csv', delimiter=',')
time, data = my_csv.transpose()

或单衬：

time, data = np.genfromtxt('./temp.csv', delimiter=',').transpose()

或另一个班轮在哪里genfromtxt换位：

time, data = np.genfromtxt('./temp.csv', delimiter=',', unpack=True)

他们更优雅吗？这取决于读者。