计算 A 列中相同值的 B 列 value_counts

Question

对于下面的输入数据框，我们要创建两列，A_count 和 B_count。

有一个输入变量称为around_row_num。 IE。对于around_row_num=2，我们要查看当前行之前的2行和之后的2行，并计算与当前行具有相同'ID'的'Label'列中'A'的出现次数。

输入：

import pandas as pd

df = pd.DataFrame({'ID': [1, 1, 1, 1, 2, 2],'Label': ['A', 'A', 'B', 'B', 'A', 'B']})
    
   ID Label
0   1     A
1   1     A
2   1     B
3   1     B
4   2     A
5   2     B

around_row_num=2 的输出：

        ID  Label A_count B_count 
        1   A     2        1
        1   A     2        1
        1   B     1        2
        1   B     0        2        
        2   A     1        1
        2   B     1        1       

谢谢！

Answer 1

从 ID 组更新到帐户：

df.set_index([df['ID'], 'Label'], append=True)['ID'].unstack()\
  .groupby('ID', as_index=False, group_keys=False)\
  .rolling(3, center=True).count()

输出：

Label    A    B
  ID           
0 1    2.0  0.0
1 1    2.0  1.0
2 1    1.0  2.0
3 1    0.0  2.0
4 2    1.0  1.0
5 2    1.0  1.0

---

试试这个：

df.set_index('Label', append=True)['ID'].unstack().rolling(3, center=True).count()

输出：

Label    A    B
0      2.0  0.0
1      2.0  1.0
2      1.0  2.0
3      1.0  2.0
4      1.0  2.0
5      1.0  1.0