我正在尝试将pandas df写入csv.但是,我需要循环很多才能以块的形式获取输出.我想在csv中垂直堆叠输出.因此,在每次迭代之后,我想写一组特定的行(1:10,然后是11:20,等等……)并清除内存,以免制作一个巨大的物体.是否可以使用df.to_csv来执行此操作?
解决方法:
import numpy as np
# df.shape returns the dimensions in a tuple, the first dimension is the number of rows
df_list = np.array_split(df, df.shape[0]/10)
for d in df_list:
d.to_csv(csv_path, mode='a')