下面是一个示例代码,演示如何遍历数据框并将数据导出为CSV文件:
import pandas as pd
# 创建示例数据框
data = {'Name': ['John', 'Emma', 'Kelly'],
'Age': [25, 30, 28],
'Country': ['USA', 'UK', 'Australia']}
df = pd.DataFrame(data)
# 遍历数据框并导出数据
for index, row in df.iterrows():
filename = f"{row['Name']}.csv" # 根据姓名创建文件名
row_data = row[['Age', 'Country']] # 选择要导出的列
row_data.to_csv(filename, index=False) # 导出数据到CSV文件
这段代码首先创建了一个示例数据框,其中包含姓名、年龄和国家三列。然后,使用iterrows()
方法遍历数据框。在每次迭代中,根据姓名创建一个文件名,并选择要导出的列。最后,使用to_csv()
方法将数据导出到CSV文件中,其中index=False
参数用于禁止导出行索引。
请注意,此示例假设数据框中的姓名列是唯一的,因此可以使用它来创建文件名。如果姓名列不唯一,您可能需要使用其他列或其他方法来创建文件名。
上一篇:遍历数据框并创建新列