자동 팀 배정 프로그램을 만들어 보았다

# 이름 통일 시킨다.
def change_A(time_list):
	result = []
	for time in time_list:
		if time == 'A 10시':
			result.append('A 새벽 10시')
		else:
			result.append(time)
	return result

# 파일 -> 결과정보, 시간정보, 사람정보, 평균인원수
def 파일가져오기(파일이름):
	df = pd.read_excel(파일이름)
	df = df[~df.duplicated(subset='이름', keep='last')]
	print("column_names: ",df.columns.tolist())
	df = df.rename(columns={'시간 ': '시간'})
	df['시간'] = df['시간'].str.split(', ')
	df['시간'] = df['시간'].apply(change_A)
	df_exploded = df.explode('시간')[['시간','이름']]
	
	unique_times = df_exploded['시간'].unique()
	df_result = pd.DataFrame()
	df_result['시간'] = unique_times
	df_result['이름'] = [[] for _ in range(len(df_result))]
	result_dict = df_result.set_index('시간')['이름'].to_dict()
	
	time_df = df_exploded.groupby('시간').agg(lambda x: x.unique().tolist() if x.dtype == 'O' else x.tolist()).reset_index()
	time_dict = time_df.set_index('시간')['이름'].to_dict()
	
	people_dict = df.set_index('이름')['시간'].to_dict()
	average_number = int(df.shape[0]/df_result.shape[0])
	
	return result_dict,time_dict,people_dict,average_number

def pop_person(name):
	for key in time_dict:
	if name in time_dict[key]:
		time_dict[key].remove(name)
	if name in people_dict:
		people_dict.pop(name)

def pop_time(time):
	for key in people_dict:
		if time in people_dict[key]:
			people_dict[key].remove(time)
		if time in time_dict:
			time_dict.pop(time)

def 어쩔수없는사람배정():
	# 어쩔수 없는 사람배정
	배정완료된사람들 = []
	배정완료된시간들 = []
	for key in people_dict: 
		if(len(people_dict[key]) == 1):
			time = people_dict[key][0]
			result_dict[time].append(key)
			배정완료된사람들.append(key)
			if(len(result_dict[time]) >= average_number):
				배정완료된시간들.append(time)
	사람과시간제거하기(배정완료된사람들,배정완료된시간들)

def 어쩔수없는시간배정():
	# 어쩔수 없는 시간배정
	배정완료된사람들 = []
	배정완료된시간들 = []
	for key in time_dict:
		# 남아있는 카드를 모두 써야 할때
		if(len(time_dict[key]) <= (average_number - len(result_dict[key]))):
			print('어쩔수없는시간배정 실행', key,": ",time_dict[key])
			result_dict[key].extend(time_dict[key])
			배정완료된사람들.extend(time_dict[key])
			배정완료된시간들.append(key)
	
	# Use Counter to find duplicated values
	counter = Counter(배정완료된사람들)
	# Get duplicated values
	duplicates = [item for item, count in counter.items() if count > 1]
	if len(duplicates) > 0:
		print("중복배정된 사람들: ", duplicates)
	
	# Get unique values
	배정완료된사람들 = list(counter.keys())
	
	사람과시간제거하기(배정완료된사람들,배정완료된시간들)

def 랜덤시간배정(time):
	배정완료된사람들 = []
	배정완료된시간들 = []
	names = random.sample(time_dict[time], average_number - len(result_dict[time]))
	result_dict[time].extend(names)
	배정완료된시간들.append(time)
	배정완료된사람들.extend(names)
	사람과시간제거하기(배정완료된사람들,배정완료된시간들)

def 남은사람배정(data):
	print('---------------------------')
	print("남은사람들 배정: ",len(data),"명")
	already_assignmed_time = []
	for name in data:
		times = list(set(data[name])- set(already_assignmed_time))
		if(len(times) != 0):
			random_time = random.choice(times)
			already_assignmed_time.append(random_time)
		else:
			random_time = random.choice(data[name])
		print(name,': ',random_time)
		result_dict[random_time].append(name)
	print("-----------배정완료------------")

def 배정하기(file_name):
	deep_copy = copy.deepcopy(people_dict)
	print('---------- #0 -----------')
	어쩔수없는사람배정()
	어쩔수없는시간배정()
	총사람수(result_dict)
	for i in range(len(time_dict)):
		print('-----------#{}------------'.format(i+1))
		랜덤시간배정(sort_dict(time_dict)[0])
		총사람수(result_dict)
		if(len(time_dict) == 0):
			break
	
	remained_people_dict = {key: deep_copy[key] for key in list(people_dict.keys())}
	남은사람배정(remained_people_dict)
	
	for time in result_dict:
		print(time , ": ",len(result_dict[time]),'명 -',result_dict[time])
	
	pd.DataFrame.from_dict(result_dict, orient='index').to_excel(file_name)