How to add a column with missing values in pandas in python

Question

I have a pandas dataframe as follows.

   name    x1     x2
0  a       5      5
1  b       8      6
2  e       3      8

I also have a list of namesas follows.

lst = [a, e]

I want to add a column named as final_values to my current dataframe where;

So, my updated dataframe should look as follows.

   name    x1     x2    final_values
0  a       5      5        1
1  b       8      6        0
2  e       3      8        1

Is there any easy way of doing this in pandas?

I am happy to provide more details if needed.

jezrael · Accepted Answer · 2019-03-26 13:22:38Z

2

Check membership by Series.isin and cast boolean to integer for True/False to 1/0 map:

df['final_values'] = df['name'].isin(lst).astype(int)

df['final_values'] = np.where(df['name'].isin(lst), 1, 0)

answered Mar 26, 2019 at 13:22

jezrael

868k103 gold badges1.4k silver badges1.3k bronze badges

Sign up to request clarification or add additional context in comments.

1 Answer 1