python - 从字符串中选择数据框行的条件

Question

我有一个这样的数据框:

data = [[1, 0, 1],
        [1, 1, 1],
        [0, 0, 1]]

df = pd.DataFrame(data=data, columns=['col1', 'col2', 'col3'])

并从这样的字符串:

cond_str = ['col1 >= 1', 'col3 >= 1']

我想实现这一目标:

df.loc[(df['col1'] >= 1) & (df['col3'] >= 1)]

有什么简单的方法可以做到这一点，还是我必须为此编写自己的解析函数？

Answer 1

您可以使用 query :

df.query("&".join(cond_str))

这导致

   col1  col2  col3
0     1     0     1
1     1     1     1