python pandas 标志如果列中的每个值有多个唯一行

Question

在下面的 DataFrame 中，我有三列:

   Code      |   Category  |    Count
     X               A          89734
     X               A          239487
     Y               B          298787
     Z               B          87980
     W               C          098454

我需要添加一列，如果一个类别有多个唯一代码(如上例中的 B)，它会得到一个标志，表示它是一个测试。

所以我正在寻找的输出是这样的:

   Code      |   Category  |    Count    | Test_Flag
     X               A          89734       
     X               A          239487
     Y               B          298787         T
     Z               B          87980          T
     W               C          098454

Answer 1

您也可以选择 transform与 numpy.where用于填充值。

df['Test_flag'] = np.where(df.groupby('Category').Code.transform('nunique') > 1, 'T', '')


>>> df
  Category Code   Count Test_flag
0        A    X   89734          
1        A    X  239487          
2        B    Y  298787         T
3        B    Z   87980         T
4        C    W   98454