讀取文本文件按列並在列表存儲在python

Question

我有以下格式的文本文件：

a,b,c,d, 
1,1,2,3, 
4,5,6,7, 
1,2,5,7, 
6,9,8,5, 

我怎麼能讀入一個列表有效，從而得到以下 輸出？

list=[[1,4,1,6],[1,5,2,9],[2,6,5,8],[3,7,7,5]] 

Answer 1

讓我們假設該文件名爲spam.txt：

$ cat spam.txt 
a,b,c,d, 
1,1,2,3, 
4,5,6,7, 
1,2,5,7, 
6,9,8,5,  

使用list comprehensions和zip()內置的功能，你可以寫一個程序，如：

>>> with open('spam.txt', 'r') as file: 
...  file.readline() # skip the first line 
...  rows = [[int(x) for x in line.split(',')[:-1]] for line in file] 
...  cols = [list(col) for col in zip(*rows)] 
... 
'a,b,c,d,\n' 
>>> rows 
[[1, 1, 2, 3], [4, 5, 6, 7], [1, 2, 5, 7], [6, 9, 8, 5]] 
>>> cols 
[[1, 4, 1, 6], [1, 5, 2, 9], [2, 6, 5, 8], [3, 7, 7, 5]] 

此外，zip(*rows)是基於在unpacking argument lists，解壓縮列表或元組，以便其元素可以作爲單獨的位置參數傳遞給函數。換句話說，zip(*rows)減少到zip([1, 1, 2, 3], [4, 5, 6, 7], [1, 2, 5, 7], [6, 9, 8, 5])。

編輯：

這是基於NumPy的參考版本：

>>> import numpy as np 
>>> with open('spam.txt', 'r') as file: 
...  ncols = len(file.readline().split(',')) - 1 
...  data = np.fromiter((int(v) for line in file for v in line.split(',')[:-1]), int, count=-1) 
...  cols = data.reshape(data.size/ncols, ncols).transpose() 
... 
>>> cols 
array([[1, 4, 1, 6], 
     [1, 5, 2, 9], 
     [2, 6, 5, 8], 
     [3, 7, 7, 5]]) 

Answer 2

可以嘗試以下代碼：

from numpy import* 

x0 = [] 
for line in file('yourfile.txt'): 
    line = line.split() 
    x = line[1] 
    x0.append(x) 

for i in range(len(x0)): 
print x0[i] 

在此，第一列被附加到X0 []。您可以以類似的方式追加其他列。