从熊猫系列中获取字典

时间:2018-03-17 12:53:00

标签: python pandas

我需要在pandas中编写一个函数,它将两个系列作为参数:一个是几个学生的成绩,另一个是他们的性别。作为一个回报,我需要获得一个字典,其中包含“男性”或“女性”的关键字和价值,这些数字是成绩超过性别的方式...例如,如果

grades = pd.Series([5, 4, 3, 5, 2]) 

genders = pd.Series(["female", "male", "male", "female", "male"]) 

该函数必须返回一个字典

{'male': 3.0, 'female': 5.0}

我知道我可以使用这两个系列的列创建一个数据框,然后使用groupby和to_dict()但不知道如何...

2 个答案:

答案 0 :(得分:3)

使用

grades.groupby(genders).mean().to_dict()

答案 1 :(得分:1)

使用

In [96]: grades.groupby(genders).mean().to_dict()
Out[96]: {'female': 5L, 'male': 3L}