使用Python中的循环将多个数据帧保存到CSV

Question

我正在尝试使用 pandas 将多个数据帧保存到 csv 循环中，同时保留数据帧的名称。

import pandas as pd
 
df1 = pd.DataFrame({'Col1':range(1,5), 'Col2':range(6,10)})
df2 = pd.DataFrame({'Col1':range(1,5), 'Col2':range(11,15)})

frames = [df1,df2]

for data in frames:
    data['New'] = data['Col1']+data['Col2']
    
for data in frames:
    data.to_csv('C:/Users/User/Desktop/{}.csv'.format(data))

这不起作用，但我正在寻找的结果是将两个数据帧以 CSV 格式保存到我的桌面。

df1.csv
df2.csv

谢谢。

Answer 1

您只需设置CSV文件的名称即可；像这样：

names = ["df1", "df2"]
for name, data in zip(names, frames):
    data.to_csv('C:/Users/User/Desktop/{}.csv'.format(name))