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

Last update on December 21 2024 07:55:36 (UTC/GMT +8 hours)

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]

Python Code Editor:

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

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.