基于布尔条件的切片熊猫DF

Question

我有两个pandas df看起来喜欢：

In [70]: print(r_ret.tail(2))
                 VTI       VOO       VGK       IEV       EWJ
2017-06-22  0.006919  0.028784 -0.269155 -0.707069 -0.223649
2017-06-23  0.089670  0.100813 -0.092122 -0.549094 -0.016048
In [71]: print(rnk.tail(2))
            VTI  VOO  VGK  IEV  EWJ
2017-06-22  2.0  1.0  4.0  5.0  3.0
2017-06-23  2.0  1.0  4.0  5.0  3.0

我想检查两个条件：

rnk < 4 and r_ret > 0

所需的输出：

2017-06-22  VTI  VOO
2017-06-23  VTI  VOO

基本上希望选择三个资产，其R_RET为> 0的最高R_RET，如果没有RNK DF的有效方法可以更好。

Answer 1

您无需使用apply就可以做到这一点，这应该可以节省大于几行的数据框架：

pd.DataFrame(np.sort(((r_ret > 0)
                      & (r_ret.rank(axis=1, ascending=False) < 4)).values
                      * r_ret.columns.values)[:, -3:],
             index=r_ret.index)
Out[264]: 
           0    1    2
2017-06-22    VOO  VTI
2017-06-23    VOO  VTI

Answer 2

我认为您可以使用apply与每一行进行比较：

mask = (rnk < 4) & (r_ret > 0)
df = mask.apply(lambda x: ', '.join(x.index[x]), axis=1) 
         .rename_axis('date').reset_index(name='val')
print (df)
         date       val
0  2017-06-22  VTI, VOO
1  2017-06-23  VTI, VOO

Answer 3

使用unstack和apply：

df[(df > 0) & (df.rank(axis=1, ascending=False) < 4)].unstack().dropna().rename_axis(['symbol', 'date']).reset_index().groupby(['date'])['symbol'].apply(lambda x: ' '.join(x))