在 Python 中使用“np.array_split”后将数据帧导出为单独的文件

Question

如何在使用 np.array_split 后导出具有不同名称的每个数据框？

我将我的数据帧分成多个部分，比方说，4，我想将它们导出为具有相应名称的单独数据帧，例如，df_split_1.csv、df_split_2.csv、df_split_3.csv, df_split_4.csv 等等。

显然，我可以使用 df_split[1].to_csv(r'W:\...\df_split_1.csv') 的方法来做到这一点，但是，如果我有 100 个这样的数据帧，那么对每个数据帧和其中一个都执行此操作并不是一个长期的解决方案。所以，问题是如何拆分数据框并将它们导出为单独的文件？

我的猜测是创建一个自动保存文件的循环，但我还没有弄清楚。

import numpy as np
import pandas as pd
    
df = pd.read_file(r'W:\...\dataframe.csv')
df_split = np.array_split(df, 4)

df_split[0].to_csv(r'W:\...\df_split_1.csv')
df_split[1].to_csv(r'W:\...\df_split_2.csv')
df_split[2].to_csv(r'W:\...\df_split_3.csv')
df_split[3].to_csv(r'W:\...\df_split_4.csv')

Answer 1

循环 enumerate:

for i, v in enumerate(df_split, 1):
    v.to_csv(fr'W:\...\df_split_{i}.csv')

在 Python 中使用“np.array_split”后将数据帧导出为单独的文件

Export dataframes as separate files after using `np.array_split` in Python

python

split

dataframe

pandas