将 pandas 数据帧拆分为 N 个块[关闭]

我目前正在尝试将 pandas 数据帧拆分为未知数量的包含每 N 行的块。

我尝试过使用 numpy.array_split() 这个函数，但是将数据帧分割成包含未知行数的 N 个块。

有没有一种巧妙的方法将 python 数据帧拆分为多个数据帧，每个数据帧包含父数据帧中特定数量的行

你可以试试这个：

def rolling(df, window, step):
    count = 0
    df_length = len(df)
    while count < (df_length -window):
        yield count, df[count:window+count]
        count += step

Usage:

for offset, window in rolling(df, 100, 100):
    # |     |                      |     |
    # |     The current chunk.     |     How many rows to step at a time.
    # The current offset index.    How many rows in each chunk.
    # your code here
    pass

还有一个更简单的想法：

def chunk(seq, size):
    return (seq[pos:pos + size] for pos in range(0, len(seq), size))

Usage:

for df_chunk in chunk(df, 100):
    #                     |
    #                     The chunk size
    # your code here

顺便提一句。所有这些都可以通过搜索在 SO 上找到。