CODEWITHSUNDEEP
pythonpandasstring
denpy
denpy

Reputation: 279

Replace values in multiple columns based on condition

data = {'Cat':  ['A','A','B','B','B','B'],
        'L1': ['0','0','0','0','0','0'],
        'L2': ['0','0','0','0','0','0'],
        'L3': ['0','0','0','0','0','0'],
        }

df = pd.DataFrame (data, columns = ['Cat','L1','L2','L3'])

Where 'Cat' is B, I would like to replace values in L1, L2, L3 to nan or ''.

df[['L1','L2','L3']] = np.where(df[['Cat'] == 'B', '')

the above did not work for me as I don't have y. Any suggestion?

Upvotes: 3

Views: 105

Answers (1)

Andrej Kesely
Andrej Kesely

Reputation: 195408

df.loc[df.Cat == "B", "L1":"L3"] = np.nan
print(df)

Prints:

  Cat   L1   L2   L3
0   A    0    0    0
1   A    0    0    0
2   B  NaN  NaN  NaN
3   B  NaN  NaN  NaN
4   B  NaN  NaN  NaN
5   B  NaN  NaN  NaN

Upvotes: 3

Related Questions