python - Pandas :groupby 然后检索 IQR

Question

我对 Pandas 很陌生，我正在尝试做以下事情:

我有两个数据框 comms 和 arts 看起来像这样(除了它们比其他列更长)

通讯:

ID    commScore           
10       5                
10       3                  
10      -1                 
11       0                
11       2              
12       9      
13      -2     
13      -1     
13       1      
13       4

艺术:

ID    commNumber
10        3 
11        2    
12        1
13        4      

我需要按 ID 对 comms 进行分组，然后将四分位数范围 (IQR) 每个 ID 的 commScore 分布。

我已经尝试过使用 groupby、agg 和 map ，但是由于我对 pandas 的看法是非常有限，我就是做不到我想要的。

有没有人有解决办法？

谢谢

安德里亚

Answer 1

我们可以使用函数 iqr 按 ID 对数据帧进行分组 并聚合列 commScore从 scipy.stats 计算四分位数间距，然后 map 这个计算的 iqr 范围在 ID 的列上艺术数据框

from scipy.stats import iqr

arts['IQR'] = arts['ID'].map(comms.groupby('ID')['commScore'].agg(iqr))

---

   ID  commNumber  IQR
0  10           3    3
1  11           2    1
2  12           1    0
3  13           4    3