在 pandas dataframe 的非常复杂的条件下获取最大值和最小值

Question

我有一个 dataframe 例如：

A       B       C
1.11    NaN     1.11
1.15    0.48    0.79
1.05   -0.29    1.26
0.99    2.90    1.13
0.90   -0.03    0.55
0.80    0.29    1.45
0.75    NaN     NaN
0.73    NaN     NaN
0.77    NaN     NaN
0.81    NaN     NaN
0.85    NaN     NaN

我想获得 4 个值并制作一个新的 dataframe。

(AA) 如果 B 列 > 0 且 C < 1 列，则在满足条件的值中获取最小值 C。

(BB) 如果 B 列 < 0 且列 C < 1，则在满足条件的值中获取最小值 C。

(CC) 如果 B 列 > 0 且 C > 1 列，则在满足条件的值中获取最大值 C。

(DD) 如果 B 列 < 0 且 C > 1 列，则在满足条件的值中获取最大值 C。

我想要的结果如下：

AA      BB      CC      DD
0.79    0.55    1.45    1.26

我将向您展示 dataframe 的另一个示例：

A       B       C
4.56    NaN     1.77
4.15    0.91    1.17
4.20    0.09    1.14
4.41    0.08    1.07
4.67    0.19    0.93
4.85    0.19    0.82
5.45    0.43    0.52
NaN     NaN     NaN
NaN     NaN     NaN
NaN     NaN     NaN
NaN     NaN     NaN

我想要的结果如下：

AA      BB      CC      DD
0.52    NaN     1.17    NaN

我怎样才能做到这一点？

Answer 1

将掩码、链掩码中的值&按位 AND 与DataFrame.loc进行过滤列C并获取min和max ，最后通过构造函数创建DataFrame ：

m1 = df.B > 0
m2 = df.C < 1
m3 = df.B < 0
m4 = df.C > 1

AA = df.loc[m1 & m2, 'C'].min()
BB = df.loc[m3 & m2, 'C'].min()
CC = df.loc[m1 & m4, 'C'].max()
DD = df.loc[m3 & m4, 'C'].max()

df = pd.DataFrame([[AA, BB, CC, DD]], columns=['AA','BB','CC','DD'])
print (df)
     AA    BB    CC    DD
0  0.79  0.55  1.45  1.26

Output 用于第二个DataFrame ：

print (df)
     AA  BB    CC  DD
0  0.52 NaN  1.17 NaN