如何在迭代生成DataFrame时将其分别保存为不同.pkl文件？

嘿，这个场景太常见了！你现在遇到的问题就是固定文件名导致每次迭代都覆盖了之前的文件，只要动态生成带唯一标识的文件名就能解决，给你几种实用的实现方式：

方式1：用迭代索引命名（最直接） #

用enumerate()获取循环的索引，把索引嵌入文件名，这样每个文件都会有唯一的序号后缀：

import pandas as pd

# 假设你的文件夹列表是folders，替换成你实际的文件夹集合
folders = ["dir_a", "dir_b", "dir_c"]

# 遍历文件夹，同时拿到索引i
for i, folder in enumerate(folders):
    # 这里替换成你生成DataFrame的实际逻辑
    df = pd.DataFrame({"content": [f"from {folder}"]})
    
    # 用f-string动态生成文件名，i从0开始递增
    df.to_pickle(f"df_{i}.pkl")

执行后你会得到df_0.pkl、df_1.pkl、df_2.pkl，每个对应循环里的一个DataFrame，不会互相覆盖。

方式2：用文件夹名称命名（更直观） #

如果想一眼看出文件对应哪个文件夹，可以直接把文件夹名称作为文件名的一部分：

for folder in folders:
    df = pd.DataFrame({"content": [f"from {folder}"]})  # 生成df的逻辑
    df.to_pickle(f"df_{folder}.pkl")

这样生成的文件是df_dir_a.pkl、df_dir_b.pkl，后期查找对应文件更方便。

方式3：自定义编号规则 #

如果你的迭代序号不是从0开始，或者有特定的编号需求，比如从1开始计数，就用自定义的计数器变量：

file_count = 1  # 自定义起始编号
for folder in folders:
    df = pd.DataFrame({"content": [f"from {folder}"]})
    df.to_pickle(f"df_{file_count}.pkl")
    file_count += 1  # 每次迭代后编号+1

小提示 #

除了f-string，你也可以用字符串的format()方法实现动态命名，比如：

df.to_pickle("df_{}.pkl".format(i))

效果和f-string完全一样，选你习惯的写法就行。

内容的提问来源于stack exchange，提问作者Marlon Teixeira