Python: replace value from dataframe if same value is present in list

Question

I have a dataframe and a list. The dataframe consists of the column 'CODE' and 'distance'. The list consists of a row of values. I want the value in 'CODE' changed to zero if values from the list are also present in the distance column of the dataframe.

import pandas as pd
df = pd.DataFrame({'CODE': np.arange(0, 220), 
                   'distance': np.arange(0, 1100, 5)})

a = [1080, 1090] #list

I can do the above operation comparing the dataframe with a single value using the code below.

df.loc[df.distance == 1080, 'CODE'] = 0

This results in the following outcome:

     CODE  distance
3       3        15
4       4        20
..    ...       ...
215   215      1075
216     0      1080
217   217      1085
218   218      1090

However, when I try to replace the value 1080 with the list a (see below) it doesn't work. How can I solve this?

df.loc[df.distance == a, 'CODE'] = 0

Djaouad · Accepted Answer · 2021-12-09 19:05:13Z

4

You can use pandas.Series.isin:

df.loc[df.distance.isin(a), 'CODE'] = 0

answered Dec 9, 2021 at 19:05

Djaouad

22.8k4 gold badges37 silver badges57 bronze badges

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

Comments

user7864386 · Accepted Answer · 2021-12-09 19:02:42Z

1

You can apply a function to df that checks if distance is in a and if it is, change the corresponding CODE to 0:

df['CODE'] = df.apply(lambda x: 0 if x['distance'] in a else x['CODE'], axis=1)

answered Dec 9, 2021 at 19:02

user7864386

Collectives™ on Stack Overflow

Python: replace value from dataframe if same value is present in list

2 Answers 2

Comments

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related