如何根据某些条件从 dataframe 创建过滤条件？

Question

我需要根据 dataframe 中的某些条件创建过滤条件。我卡在这里了。 你能帮我解决我的问题吗？ 提前致谢！

例子：

Df:

   0          1
0 kol_id    101152
1 jnj_id    7124166
2 thrc_cd   VIR
3 operator  <=
4 start_dt  05/10/2018

使用上面的 dataframe 我需要创建以下过滤器查询：

kol_id = '101152' and jnj_id = '7124166' and thrc_cd = 'VIR' and start_dt <= '05/10/2018'

如果运算符将出现在任何行中，则意味着该运算符值将在下一行过滤条件中使用，否则只有“=”运算符将在所有过滤条件中使用。

Answer 1

您可以在 pandas 中使用掩码和移位：

假设您有以下内容：

df = pd.DataFrame({'col1': ['kol', 'jnj', 'thrc', 'operator', 'start'], 'col2': [100, 200, 'VIR', '<=', '05/10/2018']})

In [10]: df                                                                                      
Out[10]: 
       col1        col2
0       kol         100
1       jnj         200
2      thrc         VIR
3  operator          <=
4     start  05/10/2018

当您屏蔽col1不是operator的每一行时，然后向底部移动然后填充您得到的默认运算符：

df['operator'] = df.mask(df['col1']!='operator').shift()['col2'].fillna('==')            

In [14]: df                                                                                     
Out[14]: 
       col1        col2 shifted
0       kol         100      ==
1       jnj         200      ==
2      thrc         VIR      ==
3  operator          <=      ==
4     start  05/10/2018      <=

现在您可以删除 col1 中包含operator的行。

并连接列以获得您的条件：

In [17]: df['res'] = '(' + df['col1'] + df.shifted + df['col2'].astype(str) + ')'                

In [18]: df                                                                                      
Out[18]: 
       col1        col2 shifted                  res
0       kol         100      ==           (kol==100)
1       jnj         200      ==           (jnj==200)
2      thrc         VIR      ==          (thrc==VIR)
3     start  05/10/2018      <=  (start<=05/10/2018)

最后加入你的列res来获取你的条件

In [19]: ' and '.join(df.res)                                                                    
Out[19]: '(kol==100) and (jnj==200) and (thrc==VIR) and (start<=05/10/2018)'