pandas apply function to column based on another column code example

Example 1: change column value based on another column pandas

# Changes the 'is_electric' column based on value in the 'type' column
# If the 'type' column == 'electric' then the 'is_electric' becomes 'YES'
df['is_electric']= df['type'].apply(lambda x: 'YES' if (x == 'electric') else 'NO')

Example 2: python pandas apply function to one column

df['a'] = df['a'].apply(lambda x: x + 1)