Python：從多個文件讀取數據到2D numpy數組或列表的方法

Question

我有n文件數量，我想從m中讀取數據來自每個文件中特定行的列數。我想以，m列表或數組結尾，其中列表或數組的每一行都是來自其中一個文件的行。什麼是做到這一點的方法？或者，這甚至是最好的辦法嗎？我可以定義不同的列表或數組（每個列表或數組都有相似的名稱，可能只是用數字索引），但處理幾個不同的列表或數組似乎通常更復雜和更復雜，而不是僅處理一個具有兩個維度的列表或數組。

例如...

file_names = ("file1" "file2" "file3" "file4") # so in this case n is 4 
for name in file_names do: 
    ### somehow save the m columns as a row in an array or list 
print resulting_list_or_array 

一些可能相關的注意事項：m列我想從文件中讀取不在文件中的第一列m（比方說我需要跳過skip_num列，而我的確知道什麼n和m是。

另外，我想讀的行不是第一行，而是說了，row_num行。

Answer 1

您可以用做一個通用的代碼內置模塊glob和numpy.genfromtxt，如下圖所示，其中的許多test*.txt文件第二條線將建成一個2-d陣列的例子在結尾處：

from glob import glob 

import numpy as np 

def read(fname, skip_header=1): 
    with open(fname) as f: 
     for i in range(skip_header): 
      f.next() 
     return f.next() 

a = np.genfromtxt((read(fname) for fname in glob('test*.txt'))) 

Answer 2

您的問題似乎明確定義

result = [] 

file_names = ("file1", "file2", "file3", "file4") 
for name in file_names do: 
    f = open(name) 
    for i, line in enumerate(f): 
     if i == 122: # note this will be line_no + 1 
      data = line.strip().split() 
      result.append([data[0], data[5], data[6], data[9]]) 

print result # a dynamically created 2D array