Python: I want to add a column to a df where if a certain value is 0, return 1 else return the original value of the column

Question

the Python code with which I am trying to achieve this result is:

df[\'column2\'] = np.where(df[\'column1\'] == 0, 1, df[\'column1\'])

Thanks in