python - 在 Pandas 中将一个 Excel 文件拆分为多个具有特定行数的文件

Question

假设我有一个 101 的 excel 文件行，我需要拆分并写入 11具有等效行号的 excel 文件 10对于每个新文件，除了最后一个，因为只剩下一行了。

这是我试过的代码，但我得到 KeyError: 11 :

df = pd.DataFrame(data=np.random.rand(101, 3), columns=list('ABC'))
groups = df.groupby(int(len(df.index)/10) + 1)
for i, g in groups:
    g.to_excel("%s.xlsx" % i, index = False, index_lable = False)

有人可以帮助解决这个问题吗？非常感谢。

引用相关: Split pandas dataframe into multiple dataframes with equal numbers of rows

Answer 1

我想你需要 np.arange :

df = pd.DataFrame(data=np.random.rand(101, 3), columns=list('ABC'))
groups = df.groupby(np.arange(len(df.index))//10)
for i, g in groups:
    print (g)