Using apply method, we can invoke given function on each value of a Series. Following snippet convert the values in Name and City columns to upper case using apply method.
columns_to_update = ['Name', 'City']
for column in columns_to_update:
df[column] = df[column].apply(to_upper)
'df[column].apply(to_upper)' statement invoke the function 'to_upper' on values of Series.
Find the below working application.
apply_transformation_on_column.py
import pandas as pd
# Create a sample DataFrame
data = {'Name': ['Krishna', 'Sailu', 'Joel', 'Chamu', 'Jitendra', "Krishna"],
'Age': [34, 35, 29, 35, 52, 34],
'City': ['Bangalore', 'Hyderabad', 'Hyderabad', 'Chennai', 'Bangalore', 'Chennai'],
'Gender': ['Male', 'Female', 'Male', 'Female', 'Male', 'Male'],
'Rating': [39, 43, 67, 100, 41, 89]}
df = pd.DataFrame(data)
print('Original DataFrame')
print(df)
def to_upper(input_str: str):
return input_str.upper()
columns_to_update = ['Name', 'City']
for column in columns_to_update:
df[column] = df[column].apply(to_upper)
print('\nDataFrame after converting the Name and City to uppercase')
print(df)
Output
Original DataFrame
Name Age City Gender Rating
0 Krishna 34 Bangalore Male 39
1 Sailu 35 Hyderabad Female 43
2 Joel 29 Hyderabad Male 67
3 Chamu 35 Chennai Female 100
4 Jitendra 52 Bangalore Male 41
5 Krishna 34 Chennai Male 89
DataFrame after converting the Name and City to uppercase
Name Age City Gender Rating
0 KRISHNA 34 BANGALORE Male 39
1 SAILU 35 HYDERABAD Female 43
2 JOEL 29 HYDERABAD Male 67
3 CHAMU 35 CHENNAI Female 100
4 JITENDRA 52 BANGALORE Male 41
5 KRISHNA 34 CHENNAI Male 89
No comments:
Post a Comment