如何在Python中按等级在数据框中的行之间均匀地划分“供应”？

import pandas as pd

# 创建一个示例数据框
data = {'供应': ['A', 'B', 'C', 'D', 'E', 'F', 'G', 'H', 'I', 'J']}
df = pd.DataFrame(data)

# 设置划分的等级数量
levels = 3

# 计算每个等级应该有的行数
num_rows = len(df)
rows_per_level = num_rows // levels

# 计算每个等级的起始行和结束行索引
start_index = 0
end_index = rows_per_level

# 在数据框中添加一个新列，表示每行的等级
df['等级'] = pd.Series()
for level in range(levels):
    # 将当前等级的行设置为对应的等级值
    df.loc[start_index:end_index, '等级'] = level + 1
    # 更新下一个等级的起始行和结束行索引
    start_index = end_index + 1
    end_index += rows_per_level

# 输出结果
print(df)