python - Pandas:基于 2 列的条件按组进行行比较

Question

我想根据跨 2 列的条件按组进行行比较。这种情况
是:(col1(i)-col1(j))*(col2(i)-col2(j)) <= 0 ，我们将每一行 i 与列 col1 和 col2 中的行 j 进行比较。如果条件满足 全部 组中的行比较，然后为该组设置为真，否则为假。

data = {'group':['A', 'A', 'A', 'B', 'B', 'B'], 
                 'col1':[1, 2, 3, 2, 3, 1], 'col2':[4, 3, 2, 2, 3, 1]} 
df = pd.DataFrame(data)
df

带输出

A True
B False

Answer 1

您可以使用 shift 用于与下一行以及 进行比较 groupby+all 用于检查组中的所有项目是否都是 True :

cond=((df['col1']-df['col1'].shift(-1))*(df['col2']-df['col2'].shift(-1))<=0)&(df['group']==df['group'].shift(-1))
cond.groupby(df['group']).all()

group
A     True
B    False
dtype: bool