python – 根据长度将数据帧拆分为相对均匀的块

我必须创建一个函数,将提供的数据帧拆分为所需大小的块.例如,如果数据帧包含1111行,我希望能够指定400行的块大小,并获得三个尺寸为400,400和311的较小数据帧.是否有便利功能来完成这项工作？什么是存储和迭代切片数据帧的最佳方法？

示例DataFrame

import numpy as np
import pandas as pd

test = pd.concat([pd.Series(np.random.rand(1111)), pd.Series(np.random.rand(1111))], axis = 1)

解决方法:

您可以使用.groupby,如下所示.

for g, df in test.groupby(np.arange(len(test)) // 400):
    print(df.shape)
# (400, 2)
# (400, 2)
# (311, 2)