以下是一个遍历不同长度的数据框并导出为CSV的示例代码:
import pandas as pd
def export_data_frames(data_frames):
for i, df in enumerate(data_frames):
df.to_csv(f'data_frame_{i}.csv', index=False)
print(f"Exported data_frame_{i}.csv")
# 创建示例数据框
df1 = pd.DataFrame({'A': [1, 2, 3], 'B': [4, 5, 6]})
df2 = pd.DataFrame({'C': [7, 8, 9, 10], 'D': [11, 12, 13, 14], 'E': [15, 16, 17, 18]})
# 将数据框放入列表
data_frames = [df1, df2]
# 导出数据框为CSV文件
export_data_frames(data_frames)
上述代码中,首先定义了一个名为export_data_frames
的函数,该函数接受一个数据框列表作为参数。然后,使用enumerate
函数遍历数据框列表,并使用to_csv
方法将每个数据框导出为CSV文件。最后,通过调用export_data_frames
函数并传入数据框列表来执行导出操作。
请确保已经安装了pandas
库,可以使用pip install pandas
命令进行安装。