python - 如何计算 Pandas 中具有精确字符串匹配的组中的行数？

Question

我有一个数据框，我想按某列进行分组，然后计算与其他列具有精确字符串匹配的行数。假设所有数据类型都是“对象”。

在伪代码中，我正在寻找类似的东西:

df.groupby('A').filter(x['B'] == '0').size()

我想按“A”列进行分组，然后计算“B”列中与字符串“0”完全匹配的行数。

编辑:我找到了一个不优雅的解决方案:

def counter(group):
    i = 0
    for item in group:
        if item == '0':
            i = i + 1
    return i

df.groupby('A')['B'].agg(counter)

一定有更好的方法。

Answer 1

我认为您在问题中提出的解决方案没有太大问题。如果您想将其制作为单衬，您可以执行以下操作:

data = np.array(list('abcdefabc')).reshape((3, 3))
df = pd.DataFrame(data, columns=list('ABC'))
df

    A   B   C
 0  a   b   c
 1  d   e   f
 2  a   b   c

df.groupby('A').agg(lambda x:list(x).count('c'))


    B   C
A       
a   0   2
d   0   0

这样做的优点是可以提供原始数据框中每列的所有值