/Python/Pandas create column from another column

Pandas create column from another column

# Creates a new column 'blue_yn' based on the existing 'color' column
# If the 'color' column value is 'blue' then the new column value is 'YES'
df['blue_yn'] = np.where(df['color'] == 'blue', 'YES', 'NO')
# Can also do this using .apply and a lambda function
df['blue_yn']= df['color'].apply(lambda x: 'YES' if (x == 'blue') else 'NO')

Create new columns pandas from another column

def label_race (row):
   if row['eri_hispanic'] == 1 :
      return 'Hispanic'
   if row['eri_afr_amer'] + row['eri_asian'] + row['eri_hawaiian'] + row['eri_nat_amer'] + row['eri_white'] > 1 :
      return 'Two Or More'
   if row['eri_nat_amer'] == 1 :
      return 'A/I AK Native'
   if row['eri_asian'] == 1:
      return 'Asian'
   if row['eri_afr_amer']  == 1:
      return 'Black/AA'
   if row['eri_hawaiian'] == 1:
      return 'Haw/Pac Isl.'
   if row['eri_white'] == 1:
      return 'White'
   return 'Other'

df.apply(lambda row: label_race(row), axis=1)

Source: stackoverflow.com

Similar codes

2 votes224 views1 answer
Flask import jsonify
Python Flask
2 votes168 views1 answer
Save a dict to pickle
Python
0 vote162 views1 answer
Why pytest return No ModuleError
Python
1 vote162 views1 answer
Count no of time element present in a string
Python
0 vote185 views1 answer
Django.core.exceptions.ImproperlyConfigured: Application labels aren't unique, duplicates: auth
Python
0 vote194 views1 answer
Pyfiglet not coming up cmd
Python
0 vote163 views1 answer
Use nltk to remove stop words
Python
38 votes160 views4 answers
Python color text console
Python
0 vote141 views1 answer
Is python not a real programing laguage because lines dont end in ;
Python
-1 vote168 views2 answers
If __name__ == '__main__': a = int(input()) b = int(input())
Python