通过numpy读取csv数据

Question

例如，我的csv文件包含十列。是否可以将其读入两个变量 - 第一个包含1-2..9列，第二个包含最后一列loadtxt或genfromtxt？

Answer 1

a = numpy.loadtxt('blah.csv', delimiter=',')    # loads it all into one big array
a1 = a[:, :-1]    # a view of all but the last column
a2 = a[:, [-1]]   # a copy of just the last column

（或者，如果您希望上一列变量为一维，则可以使用a[:,-1]而不是a[:, [-1]]来查看它。）

Answer 2

numpy.loadtext('filename.csv', usecols(0,1,2,...,n-2))
numpy.loadtext('filename.csv', usecols = n-1)

注意：列的索引应减1，因为numpy在加载时从0开始。

来自文档https://docs.scipy.org/doc/numpy/reference/generated/numpy.loadtxt.html：

  

usecols：int或sequence，可选   要读取哪些列，其中0为第一列。例如，usecols =（1,4,5）将提取第2列，第5列和第6列。默认值为None，＆gt;会导致所有列都被读取。“

Answer 3

如果你真的不需要numpy。您可以使用pandas（http://pandas.pydata.org/）或csv（https://docs.python.org/2/library/csv.html）库。