如果基于 pandas df 中的特定值

Question

我有一个这样的熊猫数据框：

np.random.seed(0)
df = pd.DataFrame(np.random.randn(5,3), columns=list('ABC'))
df['reject'] = ['RHO','AHO','AHO','RHO','RHO']

          A         B         C reject
0  1.764052  0.400157  0.978738    RHO
1  2.240893  1.867558 -0.977278    AHO
2  0.950088 -0.151357 -0.103219    AHO
3  0.410599  0.144044  1.454274    RHO
4  0.761038  0.121675  0.443863    RHO

我想根据最后一列的最后一行进行计算。 我尝试这样做：

if 'AHO' in df.iloc[-1:,-1:]:
   print('AH0')
elif 'RHO' in df.iloc[-1:,-1:]:
   print('RH0')

但是我的条件与数据框最后一个单元格的任何值都不匹配。 我认为我没有选择此单元格的正确方法，但找不到正确的方法。

谁能帮我 ？

Answer 1

你不需要slice

if 'AHO' in df.iloc[-1,-1]:
   print('AH0')
elif 'RHO' in df.iloc[-1,-1]:
   print('RH0')