Python dataframe 数据均分

import pandas as pd
import numpy as np

# 创建一个示例DataFrame
df = pd.DataFrame({
    'A': range(10),
    'B': range(10, 20)
})



# 假设我们想要均分成3部分
n_splits = len(df)/5


# 使用numpy.array_split来均分DataFrame
dfs = np.array_split(df, n_splits)

# 输出均分后的DataFrames
for i, split_df in enumerate(dfs):
    print(f"Split {i}:")
    print(split_df)
    print()