如何使用numpy genfromtxt将csv数据转换为元组

Question

我一直在用python调用元组形式的csv文件。

我用:

csv_data = np.genfromtxt('csv-data.csv', dtype=int, delimiter=',', names=True)

而数据看起来像:(对不起，我不知道如何显示csv格式)

Trial1 Trial2 Trial3
50-------70----90
60-------70----80

(2人进行3次试验的数据)

上面的genfromtxt()代码将生成:

csv_data=[(50, 70, 90) (60, 70, 80)]

而我想用逗号将数据按人分开，如:

csv_data = [(50, 70, 90), (60, 70, 80)]

从这里开始有什么帮助吗?

Answer 1

我认为您可以使用tolist方法。

http://docs.scipy.org/doc/numpy/reference/generated/numpy.ndarray.html

将数组转换为列表。

import numpy as np
csv_data = np.genfromtxt('csv-data.csv', dtype=int, delimiter=',', names=True)
print(csv_data.tolist())

上面代码的结果是:

➜ /tmp/csv $ python3 test.py
[(50, 70, 90), (60, 70, 80)]
<class 'list'>