Pandas dataframe - 检查多行是否具有相同的值

Question

我有一个如下所示的 DataFrame：

Reference	Value
String1	1
String2	0
String3	-1
String2	1
String1	1
String3	0

每个引用可以在数据框中出现一次、两次或三次；并且可以具有相同或不同的关联值。 我想创建另一个数据框，它告诉我，对于每个参考，它们是否都具有相同的值。因此，在上面的示例中，我想得到如下内容：

参考	值
String1	Yes
String2	No
String3	No

（我以 Yes 和 No 为例，但它可以是 1/0 或其他）

我该怎么做？

我最初的想法是使用 .groupby 但后来我没有找到任何类型的聚合可以帮助我......

Answer 1

您可以使用groupby + nunique来获取每个引用的唯一值的计数。然后使用 np.where 根据唯一值的数量是否为 1 来分配“是”/“否”值：

out = df.groupby('Reference', as_index=False)['Value'].nunique()
out['Value'] = np.where(out['Value'].eq(1), 'Yes', 'No')

输出：

  Reference Value
0   String1   Yes
1   String2    No
2   String3    No