通過numpy讀取csv數據

例如，我有csv文件包含十列。是否有可能將其讀入兩個變量 - 第一個將包含1-2..9列，第二個將包含最後一列，通過loadtxt或genfromtxt？通過numpy讀取csv數據

2017-04-11 Roma Karageorgievich

你可以將它讀入一個大數組'a'（非常容易使用'loadtxt'），然後查看兩個不同的切片，即'a1 = a [:,：-1]'和'a2 = a [:, [-1]] ' – jez

你看了看進口csv模塊嗎？這個答案應該完成工作：http://stackoverflow.com/questions/24662571/python-import-csv-to-list –

@ Z.Bagley是的，但是這個是一個'numpy'問題，它已經具有將csvs直接加載到'numpy'數據結構的功能。在'csv'中使用'csv'這種情況是不必要的中間人。 –

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]]的看法。）

來源

2017-04-11 21:01:12 jez

不幸的是，我不相信'a [:, [-1]]'是一個視圖，而是一個副本。 –

@ juanpa.arrivillaga好點 – jez

是的，可能你不想看到反正。 –

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或序列，可選哪些列讀取，其中0是第一個。例如，usecols =（1,4,5）將提取第2列，第5列和第6列。默認，無，>讀取中的所有列的結果。」

來源

2017-04-11 21:02:01 boethius

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

來源

2017-04-11 21:02:09

雖然這不能回答這個問題，但我可以推薦OP看看大熊貓 - 這聽起來像是他們要做一些大熊貓非常適合的事情。 – jez

通過numpy讀取csv數據

回答

相關問題