从pandas数据帧的第一列中提取值，并在保存时将其添加到文件名中

Question

我有以下数据帧

year   city         population
2002   Chicago      100000
2002   Dallas       150000
2002   Denver       200000

我想提取"；2002"；(第一列的每行中都有一个文件的值相同(从第一列添加到文件名中，我将保存

输出文件名-2002_city_population.csv

我正在试用这个

df = pd.read_csv('city_population.csv', index_col=0)
df.to_csv('2002_city_population.csv')

目前我正在硬编码"；2002"；文件名中。但我希望2002年来自文件的第一列，因为每个文件都有不同的年份

Answer 1

您可以使用一个变量和一些f字符串格式来完成此操作。

year = df.at[0, 'year']
df.to_csv(f'{year}_city_population.csv')

Answer 2

您的数据框架中还有其他年份吗？如果是这样，你可以这样做：

for year, group in df.groupby("year"):
group.to_csv(f"{year}_city_population.csv")