Python將CSV文件分配給變量

Question

我的任務是導入一個CSV文件並將其製作成一個數組，我已經完成了。現在我的任務是將數據分配給一個變量 - 這可能嗎？例如，我將如何通過鍵入myVariableName [1]來訪問數組的第二列？

import csv 
import numpy as np 
f = open("Test.csv") 
csv_f = csv.reader(f) 
for row in csv_f: 
    print(np.array(row)) 
f.close() 

Answer 1

您可以使用numpy.loadtxt()。

a.csv：

1,2,3 
4,5,6 
7,8,9 

代碼：

import numpy as np 

matrix = np.loadtxt(open("a.csv","rb"),delimiter=",") 
print(matrix[:,1]) 
# prints [ 2. 5. 8.] 

Answer 2

創建一個列表，從第2列項目可以爲您處理CSV文件中的每一行追加。下面的代碼將包含從第2列項變量column_2：

import csv 

column_2 = [] 
f = open("Test.csv") 
csv_f = csv.reader(f) 
for row in csv_f: 
    column_2.append(row[1]) 
f.close() 
print column_2 

你可以用一個列表理解取代所有代碼：

import csv 

column_idx = 1 
with open('Test.csv') as f: 
    column_2 = [row[column_idx] for row in csv.reader(f)] 
print column_2