老公问我能不能给他写个程序,单位组织活动时,把人员随机分组。简单,我来帮帮他。
import pandas as pd
data = pd.read_excel('参加人员名单.xlsx')#导入人员花名册
data = data.sample(frac = 1,random_state = 0)#sample方法完成随机抽样,类似重新洗牌
num = data.shape[0]#看一下有多少人
team_num = 1
for i in range(0,num,5):#每五个人一组
team = data.iloc[i:i + 5]#切片分组
team = list(team['人'])#dataframe 转列表
print(f'第{team_num}组:{team}')
team_num += 1
网友评论