w3resource

Pandas: Extract the sentences where a specific word is present in a given column of a given DataFrame


38. Extract Sentences Containing Specific Word

Write a Pandas program to extract the sentences where a specific word is present in a given column of a given DataFrame.

Sample Solution:

Python Code :

import pandas as pd
import re as re
df = pd.DataFrame({
    'company_code': ['Abcd','EFGF', 'zefsalf', 'sdfslew', 'zekfsdf'],
    'date_of_sale': ['12/05/2002','16/02/1999','05/09/1998','12/02/2022','15/09/1997'],
    'address': ['9910 Surrey Avenue','92 N. Bishop Avenue','9910 Golden Star Avenue', '102 Dunbar St.', '17 West Livingston Court']
})
print("Original DataFrame:")
print(df)
def pick_only_key_sentence(str1, word):
    result = re.findall(r'([^.]*'+word+'[^.]*)', str1)
    return result
df['filter_sentence']=df['address'].apply(lambda x : pick_only_key_sentence(x,'Avenue'))
print("\nText with the word 'Avenue':")
print(df)

Sample Output:

Original DataFrame:
  company_code date_of_sale                   address
0         Abcd   12/05/2002        9910 Surrey Avenue
1         EFGF   16/02/1999       92 N. Bishop Avenue
2      zefsalf   05/09/1998   9910 Golden Star Avenue
3      sdfslew   12/02/2022            102 Dunbar St.
4      zekfsdf   15/09/1997  17 West Livingston Court

Text with the word 'Avenue':
  company_code            ...                        filter_sentence
0         Abcd            ...                   [9910 Surrey Avenue]
1         EFGF            ...                       [ Bishop Avenue]
2      zefsalf            ...              [9910 Golden Star Avenue]
3      sdfslew            ...                                     []
4      zekfsdf            ...                                     []

[5 rows x 4 columns]

For more Practice: Solve these Related Problems:

  • Write a Pandas program to extract sentences from a text column that contain a specific word using regex and then output them.
  • Write a Pandas program to filter a DataFrame column for sentences that include a target word and then list the matching sentences.
  • Write a Pandas program to split a column into sentences and then extract those containing a specific keyword.
  • Write a Pandas program to apply a regex search on a column to retrieve only the sentences where a given word appears.

Go to:


Previous: Write a Pandas program to extract only words from a given column of a given DataFrame.
Next: Write a Pandas program to extract the unique sentences from a given column of a given DataFrame.

Python Code Editor:

Have another way to solve this solution? Contribute your code (and comments) through Disqus.

What is the difficulty level of this exercise?

Test your Programming skills with w3resource's quiz.



Follow us on Facebook and Twitter for latest update.