CODEWITHSUNDEEP
pythonpandasdataframe
Pradeep M
Pradeep M

Reputation: 991

USING LIKE inside pandas.query()

I have been using Pandas for more than 3 months and I have an fair idea about the dataframes accessing and querying etc.

I have got an requirement wherein I wanted to query the dataframe using LIKE keyword (LIKE similar to SQL) in pandas.query().

i.e: Am trying to execute pandas.query("column_name LIKE 'abc%'") command but its failing.

I know an alternative approach which is to use str.contains("abc%") but this doesn't meet our requirement.

We wanted to execute LIKE inside pandas.query(). How can I do so?

Upvotes: 85

Views: 138737

Answers (6)

Abhijeet
Abhijeet

Reputation: 298

DataFrame:

    Name    Code  App

0  Jhon     8010  google
1  Michael  9020  github
2  Mandy    1240  google.com
3  Krish    1240  facebook

Search a word or related words in Dataframe

S = df[df["column_name"].str.contains("word")]
S.head()

Example:

Myword = input("Enter the word, want to search:")

S = df[df["App"].str.contains(Myword)]
S.head()

print(S)

Output:

Enter the word, want to search: google

   Name   Code  App
0  Jhon   8010  google
2  Mandy  1240  google.com

Note: This method is case sensitive

Upvotes: 0

P.Panayotov
P.Panayotov

Reputation: 573

@volodymyr is right, but the thing he forgets is that you need to set engine='python' to expression to work.

Example:

>>> pd_df.query('column_name.str.contains("abc")', engine='python')

Here is more information on default engine ('numexpr') and 'python' engine. Also, have in mind that 'python' is slower on big data.

Upvotes: 34

volodymyr
volodymyr

Reputation: 7554

If you have to use df.query(), the correct syntax is:

df.query('column_name.str.contains("abc")', engine='python')

You can easily combine this with other conditions:

df.query('column_a.str.contains("abc") or column_b.str.contains("xyz") and column_c>100', engine='python')

It is not a full equivalent of SQL Like, however, but can be useful nevertheless.

Upvotes: 106

Shovalt
Shovalt

Reputation: 6776

A trick I just came up with for "starts with":

pandas.query('"abc" <= column_name <= "abc~"')

Explanation: pandas accepts "greater" and "less than" statements for strings in a query, so anything starting with "abc" will be greater or equal to "abc" in the lexicographic order. The tilde (~) is the largest character in the ASCII table, so anything starting with "abc" will be less than or equal to "abc~".

A few things to take into consideration:

  • This is of course case sensitive. All lower case characters come after all upper cases characters in the ASCII table.
  • This won't work fully for Unicode strings, but the general principle should be the same.
  • I couldn't come up with parallel tricks for "contains" or "ends with".

Upvotes: 4

Terrance DeJesus
Terrance DeJesus

Reputation: 241

Super late to this post, but for anyone that comes across it. You can use boolean indexing by making your search criteria based on a string method check str.contains.

Example:

dataframe[dataframe.summary.str.contains('Windows Failed Login', case=False)]

In the code above, the snippet inside the brackets refers to the summary column of the dataframe and uses the .str.contains method to search for 'Windows Failed Login' within every value of that Series. Case sensitive can be set to true or false. This will return boolean index which is then used to return the dataframe your looking for. You can use .fillna() with this in the brackets as well if you run into any Nan errors.

Hope this helps!

Upvotes: 11

khammel
khammel

Reputation: 2127

Not using query(), but this will give you what you're looking for:

df[df.col_name.str.startswith('abc')]


df
Out[93]: 
  col_name
0     this
1     that
2     abcd

df[df.col_name.str.startswith('abc')]
Out[94]: 
  col_name
2     abcd

Query uses the pandas eval() and is limited in what you can use within it. If you want to use pure SQL you could consider pandasql where the following statement would work for you:

sqldf("select col_name from df where col_name like 'abc%';", locals())

Or alternately if your problem with the pandas str methods was that your column wasn't entirely of string type you could do the following:

df[df.col_name.str.startswith('abc').fillna(False)]

Upvotes: 16

Related Questions