使用 to_pickle 在循环中保存多个数据帧

Question

嗨，我有 4 个 pandas 数据框：df1、df2、df3、df4。 我喜欢做的是使用to_pickle 迭代（使用for 循环）保存此数据帧。 我所做的是这样的：

out = 'mypath\\myfolder\\'

r = [ orders, adobe, mails , sells]
for i in r:
    i.to_pickle( out +   '\\i.pkl')

该命令很好，但它不会用他的名字保存每个数据库，而是覆盖相同的数据库i.pkl（我认为是因为我的代码不正确） 似乎无法用他的名字重命名每个数据库（例如，for循环中的订单orders以名称i.pkl保存，以此类推，orders数据框涉及） 我希望保存 4 个数据框，并在对象 r 中插入名称（因此：orders.pkl、adobe.pkl、mails.pkl、sells.pkl）

我该怎么做？

Answer 1

你不能对变量名进行字符串化（这不是你通常会做的事情），但你可以做一些简单的事情：

import os

out = 'mypath\\myfolder\\'

df_list = [df1, df2, df3, df4]
for i, df in enumerate(df_list, 1):
    df.to_pickle(os.path.join(out, f'\\df{i}.pkl')

---

如果您想为文件提供自定义名称，我的建议是：使用字典。

df_map = {'orders': df1, 'adobe': df2, 'mails': df3, 'sells': df4}
for name, df in df_map.items():
    df.to_pickle(os.path.join(out, f'\\{name}.pkl')