如何将数据帧导出为 500 行块的 csv

SOK*_*SOK 1 python pandas

我有一个大型数据框,我将其导出为 CSV 并上传到第三方产品。第三方产品只能接受最多 500 行数据的上传,因此我想知道如何将数据帧导出到较小的文件中。

目前我的代码如下:

df.to_csv("Export.csv",index=False)
Run Code Online (Sandbox Code Playgroud)

但理想情况下希望代码能够导出文件,以便它给出:

Export1.csv
Export2.csv
Export3.csv
Run Code Online (Sandbox Code Playgroud)

等等,直到全部完成并且每行有 500 行(然后是剩下的)。任何帮助将非常感激!

vra*_*raj 5

  • 您可以使用df.groupby函数将 df 拆分为 500 个块。并保存每个块。您可以在此处查看参考(/sf/answers/1799212131/
import numpy as np
no_of_rows = 500
for k,g in df.groupby(np.arange(len(df))//no_of_rows):
    g.to_csv('Export{}.csv'.format(k+1), index=False)
Run Code Online (Sandbox Code Playgroud)