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

创建一个列表，当您处理CSV文件中的每一行时，可以在其中添加第2列中的项目。以下代码将包含变量column_2中第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