python - 将 pandas Dataframe 分成 n 等份 + 1

Question

我有一个包含 44150 行的 pandas Dataframe。

我想分成子数据帧，每个子数据帧包含 100 行，除了最后一个必须包含 50 行。

我尝试过使用numpy.array_split，但它会将其拆分为 392 个大小为 100 的数据帧和 50 个大小为 99 的数据帧。

有没有办法按照我想要的方式分割它？

Answer 1

您可以使用iloc和列表理解:

df = pd.DataFrame({
    'x':np.random.randn(44150),
    'y':np.random.randn(44150),
})

S = 100
N = int(len(df)/S)
frames = [ df.iloc[i*S:(i+1)*S].copy() for i in range(N+1) ]

最后一个 DataFrame - 可以在 frames[-1] 中找到 - 有 50 行，而其他 DataFrame 有 100 行。