从 CSV 文件复制列的 Pythonic 方法

Question

我正在寻找一种 pythonic 和简洁的方法来选择 .csv 文件中的列并将该列的所有单元格存储在列表中。

import csv    

with open("/path/to/file.csv","r") as csvfile:
    reader = csv.DictReader(csvfile, delimiter=";")
    # TODO: select column for key "foo"
    # TODO: select column for key "bar"
    # TODO:store "foo" data in list
    # TODO: store "bar" data in list

Answer 1

在纯 Python 中从 DictReader 行字典中获取列很简单，其他人现在可能正在为此效果写一个答案，所以我不会重复这项工作，而是在我最喜欢的一个中展示如何做到这一点用于数据操作的 Python 库，pandas :

>>> import pandas as pd
>>> df = pd.read_csv("somefile.csv", sep=";")
>>> df
   foo  bar      apple
0    1  100       pear
1    2  200     orange
2    3  300  tangerine
3    4  400      peach
>>> df["foo"]
0    1
1    2
2    3
3    4
Name: foo
>>> df["bar"]
0    100
1    200
2    300
3    400
Name: bar
>>> df["foo"] * df["bar"]
0     100
1     400
2     900
3    1600
>>> list(df["foo"] * df["bar"])
[100, 400, 900, 1600]

在pandas 之前的黑暗日子里，我有自己的手工库来进行这种数据访问。几年前用 pandas 大约十五分钟后，我把它扔了..