如何检查 pandas dataframe 列中的所有值是否相等？

Question

我有一个像这样的 dataframe

  name data result 
0  x    100
1  x    100
2  x    100
3  x    100
4  x    100
5  y    100
6  y    90
7  y    90
8  y    100
9  y    85

我想检查name列中的每个组在data列中是否具有相同的值。

所以对于每个x组，如果对应的data值都相等，则在result列中full 。 如果一个组的值不是全部相等，请在result列中写入nearly 。

我尝试将 dataframe 分组：

dfx = df.groupby('name')
dfx = dfa.get_group('x')

但它并不能真正帮助检查每个值是否相同，请写入result列。

我尝试创建一个 function 来检查唯一值

def check_identicals(row):
    if(df.sent.nunique() == 1):
        print('Full')

这里的想法是然后将 function 应用到每一行，并在result列中写入 output。

理想output：

   name data result 
0  x    100   full
1  x    100   full
2  x    100   full
3  x    100   full
4  x    100   full
5  y    100   nearly
6  y    90    nearly
7  y    90    nearly
8  y    100   nearly
9  y    85    nearly

Answer 1

使用numpy.where与GroupBy.transform和DataFrameGroupBy.nunique比较新Series中与原始DataFrame相同大小的所有值：

df['result'] = np.where(df.groupby('name')['data'].transform('nunique') == 1,'full','nearly')
print (df)
  name  data  result
0    x   100    full
1    x   100    full
2    x   100    full
3    x   100    full
4    x   100    full
5    y   100  nearly
6    y    90  nearly
7    y    90  nearly
8    y   100  nearly
9    y    85  nearly

编辑：

为了测试每组的所有缺失值是否使用numpy.select和另一个条件，将缺失值与transform和GroupBy.all进行比较：

m1 = df.groupby('name')['data'].transform('nunique') == 1
m2 = df['data'].isna().groupby(df['name']).transform('all')

df['result'] = np.select([m1, m2], ['full', 'all_missing'],'nearly')
print (df)
  name   data       result
0    x  100.0         full
1    x  100.0         full
2    x  100.0         full
3    x  100.0         full
4    x  100.0         full
5    y  100.0       nearly
6    y   90.0       nearly
7    y   90.0       nearly
8    z    NaN  all_missing
9    z    NaN  all_missing