python - 将列作为列表导入列标题变量

Question

我有一个这样的文件

AAPL, TSCO, ASDA, ....
10, 11, 12, ...
9, 10, 11, ...
...,...,...,...

现在有一种方法可以将所有这些列表作为列标题的变量导入。所以它将是AAPL = [ 10, 9, ...,...]

我知道如何使用以下方法打印列标题列表:

import pandas as pd
df = pd.read_csv('foo.csv', index_col=0)

col_headers = list(df.columns)

所以现在我可以遍历这个列表，但是有没有办法自动将变量名指定为列表标题？

Answer 1

这里最好的是创建列表字典，并为每个列表按键选择 - 这里是列名:

d = df.to_dict('list')
print (d)
{'AAPL': [10, 9], 'TSCO': [11, 10], 'ASDA': [12, 11]}

print (d['AAPL'])
[10, 9]

print (type(d['AAPL']))
<class 'list'>

你需要的是可能的，但不是recommended :

for c in df.columns:
    globals()[c] =  df[c].tolist()

print (AAPL)
[10, 9]