在for循環中創建numpy數組

Question

我希望能夠基於可能發生變化的數字創建numpy數組。
例如，假設我有50個文本文件，其中包含一組2x2數字

我想將這50個文件裝載爲numpy數組，並在代碼中稍後使用它們。代碼示例可能如下所示：

import load numpy as np 
num = 50 #this could change based on different conditions 

for i in arange(num): 
    data%d % i = np.loadtxt("datafromafile%d.txt" % i) 

這是這樣的可能嗎？ 感謝

Answer 1

您可以將它們存儲在一個列表：

data = list() 
for i in arange(num): 
    data.append(np.loadtxt("datafromafile%d.txt" % i)) 

然後你就可以訪問每個陣列：

>>> data[0] + data[1] # sum of the first and second numpy array 

Answer 2

作爲oneliner這將是：

NUM = 50 
data = [np.loadtxt("datafromafile%d.txt" % value) for value in np.arange(NUM)] 

或

FILES = ['file1', 'file2', 'file3'] 
data = {key: np.loadtxt(key) for key in FILES} 

作爲字典的文件名作爲關鍵字。