设置pandas.DataFrame对象的列

Question

理想情况下，我希望能够像：

cols = ['A', 'B', 'C']
df = pandas.DataFrame(index=range(5), columns=cols)
df.get_column(cols[0]) = [1, 2, 3, 4, 5]

这样做的pythonic / pandonic方式是什么？

编辑：我知道我可以通过df.A访问“A”列，但一般情况下我不知道列名是什么。

Answer 1

好的，这一点特别简单。

df[cols[0]] = [1, 2, 3, 4, 5] 

Answer 2

您无需单独存储DataFrame所具有的列。

您可以通过访问DataFrame.columns变量找出pandas DataFrame中存在哪些列。

要访问附加到特定列的系列，您可以使用DataFrame的[]

的getitem方法

微小的例子：

col = df.columns[0]
df[col] = [1, 2, 3, 4, 5]