如何将 pandas DataFrame 拆分为多个 DataFrame？

Question

我有一个由 231840 行组成的数据框。我需要将其分成 161 个单独的表，每个表包含 1440 行，即第一个表包含前 1440 行，第二个表包含接下来的 1440 行，依此类推，直到我得到 161 个单独的表，行数组合为231840 行。有任何想法吗？

Answer 1

您可以使用，np.array_split来分割数据框：

import numpy as np

dfs = np.array_split(df, 161) # split the dataframe into 161 separate tables

编辑（根据 df 中的顺序号分配新的列dfs）：

dfs = [df.assign(new_col=i) for i, df in enumerate(dfs, 1)]