在Pandas中根据特定条件选择行

Question

因此，我对使用Pandas数据格式非常陌生。

我有一个dataframe，其中一个列表示客户I，另一个列保存了如下所示的口味和满意度分数。

​

​

虽然每个客户应该有6行专用于它们，客户1只有5。我如何创建一个新的数据have，只打印出客户谁有6行？

我试过这样做：df['Customer No'].value_counts() == 6，但它不起作用。

Answer 1

这里有一种方法

如果您以代码(最好)或文本的形式发布数据，我将能够共享结果。

# create a temporary column 'c' by grouping on Customer No
# and assigning count to it using transform
# finally, using loc to select rows that has a count eq 6

(df.loc[df.assign(
    c=df.groupby(['Customer No'])['Customer No']
    .transform('count'))['c'].eq(6]
)