python pandas数据框在读取excel文件时添加带有值的列

Question

有没有办法在使用 python pandas 读取 excel 文件时添加具有特定值的列？

现在我有两个步骤，但我需要做 30 多个文件，所以我想找到优雅的方法来做到这一点！

1) df_2007 = pd.read_excel('may2007_dl.xls')
2) df_2007['year'] = 2007

谢谢。

Answer 1

你的代码没有任何问题。但是，如果您想在一行中完成，您可以使用assign，它将新列分配给 DataFrame 并返回一个新对象（副本），其中包含除新列之外的所有原始列。

df_2007 = pd.read_excel('may2007_dl.xls').assign(y=2007)

示范：

In [68]: pd.read_csv('/tmp/test.csv')
Out[68]: 
  Unnamed: 0  one  two
0          a  1.0  1.0
1          b  2.0  2.0
2          c  3.0  3.0
3          d  NaN  4.0

In [69]: pd.read_csv('/tmp/test.csv').assign(a=2007)
Out[69]: 
  Unnamed: 0  one  two     a
0          a  1.0  1.0  2007
1          b  2.0  2.0  2007
2          c  3.0  3.0  2007
3          d  NaN  4.0  2007