Панды добавить количество уникальных столбцов

Question

Учитывая следующий кадр данных:

import pandas as pd 
df = pd.DataFrame(
    {'A' : ['foo', 'foo', 'foo', 'foo', 
      'bar', 'bar', 'bar', 'bar'], 
    'B' : [2, 4, 4, 2, 5, 4, 3, 2]}) 
df 

    A  B 
0 foo  2 
1 foo  4 
2 foo  4 
3 foo  2 
4 bar  5 
5 bar  4 
6 bar  3 
7 bar  2 

Я хотел бы столбец («C») из подсчета уникальных значений для «B» в группе «A», как это с помощью функция лямбда x:

A  B C 
0 foo  2 2 
1 foo  4 2 
2 foo  4 2 
3 foo  2 2 
4 bar  5 1 
5 bar  4 1 
6 bar  3 1 
7 bar  2 1 

Заранее благодарен!

Answer 1

Если компьютер прямо о своей цели, может быть

>>> df["C"] = df.groupby(["A","B"])["A"].transform("count") 
>>> df 
    A B C 
0 foo 2 2 
1 foo 4 2 
2 foo 4 2 
3 foo 2 2 
4 bar 5 1 
5 bar 4 1 
6 bar 3 1 
7 bar 2 1 

даст вам то, что вы хотите? Мы группируем пары (A, B).

---

Cute Немного истории: это первоначально то, что я сделал, но потом я попробовал еще раз и обнаружил, что мне не нужен ["A"]. Но причина, по которой она работала во второй раз без этого, заключалась в том, что у меня был столбец C, поэтому было что-то для того, чтобы код действовал .. (sigh)