导入CSV文件并将每列设为自己的列表

Question

我有一个csv文件，其中包含具有很多行的此类数据：

2017, UNC (1), Gonzaga (1), Oregon (3), South Carolina (7)
2016, Villanova, (2), UNC (1), Oklahoma (2), Syracuse (10)

我正在尝试导入文件，并以python中最简单的方式将每一列放入自己的列表中。

Answer 1

这将为您提供列列表，每列都是项目列表。

import pandas as pd

def get_columns(csv_file):
    df = pd.read_csv(your.csv)

    columns = []
    for column in df:
        columns += column
    return columns

Answer 2

每次安装python时，都可以使用内置的csv模块标准。

import csv
header1 = []
header2 = []
# so on
with open('file.csv') as csvfile:
    reader = csv.DictReader(csvfile, fieldnames=['header1', 'header2', 'header3', 'header4', 'header5'])
    for row in reader:
        header1.append(row['header1'])
        header2.append(row['header2'])
        # so on