python - Pandas - 相当于 pandas 查询中的 str.contains()

Question

使用具有以下条件的子集创建数据框

subset_df = df_eq.loc[(df_eq['place'].str.contains('Chile')) & (df_eq['mag'] > 7.5),['time','latitude','longitude','mag','place']]

想在 Pandas 中使用 query() 复制上述子集。但是不确定如何在 Pandas 查询中复制 str.contains() 等效项。查询中的“喜欢”似乎不起作用

query_df = df_eq[['time','latitude','longitude','mag','place']].query('place like \'%Chile\' and mag > 7.5')

place like '%Chile'and mag >7.5 
            ^
SyntaxError: invalid syntax

任何帮助将不胜感激

Answer 1

我认为这里发生的是您无法使用该方法 str.contains内query Pandas 方法。您可以做的是创建一个掩码并从 query 中引用该掩码。使用 at 符号 (@)。尝试这个:

my_mask = df_eq["feature"].str.contains('my_word')
df_eq.query("@my_mask")