使用 str.contains 方法清理要在 pandas 查询中使用的输入

Question

How can I sanitize the input in the code below so that it can be used to query a dataset df ?

query = f"`{field_name}`.str.contains('''{input()}''', case=False)"
df.query(query)

The main problem with the code above is that when the input contains triple quotes or backslashes it throws an error. Also keep in mind that the dataframe also contains backslashes in some cells and thus I would like the query to be able to perform that search as well (eg if the input is a\s I would like the query to return rows that contain a\s like for example aaaa\saaaaa would be a match).

Assume that field_name is given and not going to cause trouble.

Answer 1

If I understand you correctly, you want this:

import pandas as pd
import numpy as np


s1 = pd.Series(['Mouse', 'dog  a\s', 'house and parrot', '23', np.NaN, 'aaaa\saaaaa', ' \  """   '])
s2 = s1.str.contains(input('input: '), regex=False)
print(s2)

input: a\s
0    False
1     True
2    False
3    False
4      NaN
5     True
6    False
dtype: object

Process finished with exit code 0

input: """
0    False
1    False
2    False
3    False
4      NaN
5    False
6     True
dtype: object

Process finished with exit code 0