正在将数据帧转换为数组、哈希表

Question

我有一个这种格式的数据帧。但希望将其转换为下面模型中的格式，对每个Manager进行迭代。请注意，实习生总是兼职，所有其他角色总是全职。任何帮助都将不胜感激！数据帧

实物模型

Answer 1

我认为这将对您有所帮助。

import pandas as pd
def collect_names(series):
return list(series.values)
df = pd.read_csv(r'/content/Dataframe.csv')
df = df.assign(name_with_role = lambda x: x['EMPLOYEE NAME']+' ('+ x['ROLE']+ ')')
intermediate_dict = df.groupby(by=['MANAGER NAME', 'STATUS'], as_index=False)['name_with_role'].agg(collect_names).to_dict('records')
output_dict = {}
for i in intermediate_dict:
if i['MANAGER NAME'] not in output_dict.keys():
output_dict[i['MANAGER NAME']]= {i['STATUS']: i['name_with_role']} 
else:
output_dict[i['MANAGER NAME']].update({i['STATUS']: i['name_with_role']} )
print(output_dict)