如何将 CSV 文件的列导出为 python 中的 arrays？

Question

我有一个包含 2 列的 CSV 文件，我想将这 2 列导出为 2 arrays。 如何使用 Python 完成它？

Answer 1

您可以通过这种方式使用列表理解和循环通过 csv 文件：

import csv 

with open('example.csv') as csv_file:
    csv_reader = csv.reader(csv_file, delimiter=',')
    column1 = [row[0] for row in csv_reader]
    column2 = [row[0] for row in csv_reader]

Answer 2

我会使用 pandas 以获得更好的性能和未来的操作

import pandas

df = pandas.read_csv("example.csv")

for row in df.iterrows():
    print(row)

Answer 3

您可以使用 pandas 并指定列名来执行此操作。

import pandas as pd
df=pd.read_csv("example.csv")
        
array1=df["col_name1"].values # as numpy array
array2=list(df["col_name2"].values) # as python array

Answer 4

对csv.reader返回的行的迭代使用zip将数据组装成列：

import csv
with open('example.csv') as f:
    reader = csv.reader(f)
    columns_as_lists = [list(c) for c in zip(*reader)]
print(columns_as_lists[0])  # All the values in the first column of your CSV
print(columns_as_lists[1])  # All the values in the second column of your CSV