Pandas: Check if a specified column starts with a specified string in a DataFrame

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

Write a Pandas program to check if a specified column starts with a specified string in a DataFrame.

Sample Solution:

Python Code :

import pandas as pd
df = pd.DataFrame({
    'company_code': ['Abcd','EFGF', 'zefsalf', 'sdfslew', 'zekfsdf'],
    'date_of_sale': ['12/05/2002','16/02/1999','25/09/1998','12/02/2022','15/09/1997'],
    'sale_amount': [12348.5, 233331.2, 22.5, 2566552.0, 23.0]
})
print("Original DataFrame:")
print(df)
print("\nIf a specified column starts with a specified string?")
df['company_code_starts_with'] = list(
    map(lambda x: x.startswith('ze'), df['company_code']))
print(df)

Sample Output:

Original DataFrame:
  company_code date_of_sale  sale_amount
0         Abcd   12/05/2002      12348.5
1         EFGF   16/02/1999     233331.2
2      zefsalf   25/09/1998         22.5
3      sdfslew   12/02/2022    2566552.0
4      zekfsdf   15/09/1997         23.0

If a specified column starts with a specified string?
  company_code           ...            company_code_starts_with
0         Abcd           ...                               False
1         EFGF           ...                               False
2      zefsalf           ...                                True
3      sdfslew           ...                               False
4      zekfsdf           ...                                True

[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 get the length of the integer of a given column in a DataFrame.
Next: Write a Pandas program to swap the cases of a specified character column in a given DataFrame.