CODEWITHSUNDEEP
pythonpandas
user_v27
user_v27

Reputation: 423

Column Addition if not available Pandas

I use the long list of codes similar to below codes to check & create a column with '0' value,

if a column does not exist in Data frame (df),

if 'E184' not in df: df['E184'] = 0
if 'E185' not in df: df['E185'] = 0 
if 'E187' not in df: df['E187'] = 0 
if 'E186' not in df: df['E186'] = 0 
if 'E176' not in df: df['E176'] = 0 
if 'E177' not in df: df['E177'] = 0 
if 'E181' not in df: df['E181'] = 0 
if 'E210' not in df: df['E210'] = 0 
if 'E175' not in df: df['E175'] = 0 
if 'E209' not in df: df['E209'] = 0 
if 'E174' not in df: df['E174'] = 0 
if 'E212' not in df: df['E212'] = 0 
if 'E211' not in df: df['E211'] = 0 
if 'E154' not in df: df['E154'] = 0 
if 'E219' not in df: df['E219'] = 0 
if 'E230' not in df: df['E230'] = 0 
if 'E220' not in df: df['E220'] = 0 
if 'E157' not in df: df['E157'] = 0 
if 'E188' not in df: df['E188'] = 0 
if 'E234' not in df: df['E234'] = 0

Is there a simple and efficient way to do it.

Upvotes: 2

Views: 85

Answers (3)

Pablo C
Pablo C

Reputation: 4761

You can use pandas.DataFrame.reindex:

codes = ['E184', 'E185', 'E187', 'E186', 'E176', 'E177', 'E181', 'E210', 'E175',
         'E209', 'E174', 'E212', 'E211', 'E154', 'E219', 'E230', 'E220', 'E157',
         'E188','E234']

df = df.reindex(columns = set(df.columns.to_list() + codes), fill_value = 0)

Upvotes: 4

piRSquared
piRSquared

Reputation: 294498

Setup

df = pd.DataFrame({'E001': [1, 2, 3]})

df

   E001
0     1
1     2
2     3

Create a dictionary of columns not in df

codes = ['E001', 'E002', 'E003', 'E004']

df.assign(**dict.fromkeys({*codes} - {*df}, 0))

   E001  E002  E004  E003
0     1     0     0     0
1     2     0     0     0
2     3     0     0     0

Combine with a newly constructed dataframe

codes = ['E001', 'E002', 'E003', 'E004']

df.combine_first(pd.DataFrame(0, df.index, codes))

   E001  E002  E004  E003
0     1     0     0     0
1     2     0     0     0
2     3     0     0     0

Upvotes: 3

Ch3steR
Ch3steR

Reputation: 20659

You can use pd.Index.union with df.reindex

codes = ['E184', 'E185', 'E187', 'E186', 'E176', 'E177', 'E181', 'E210', 'E175',
         'E209', 'E174', 'E212', 'E211', 'E154', 'E219', 'E230', 'E220', 'E157',
         'E188','E234']

df.reindex(df.columns.union(codes, sort=False), fill_value=0, axis=1)

Upvotes: 6

Related Questions