Pandas: How to replace Zero values in a column with the mean of that column, For all columns with Zero Value

Question

I have a dataframe with multiple values as zero. I want to replace the values that are zero with the mean values of that column Without repeating code. I have columns called runtime, budget, and revenue that all have zero and i want to replace those Zero values with the mean of that column.

Ihave tried to do it one column at a time like this:

    print(df['budget'].mean())    
    -> 14624286.0643    
    df['budget'] = df['budget'].replace(0, 14624286.0643)    

Is their a way to write a function to not have to write the code multiple time for each zero values for all columns?

Answer 1

Same we can achieve directly using replace method. Without fillna

df.replace(0,df.mean(axis=0),inplace=True)

Method info: Replace values given in "to_replace" with "value".

Values of the DataFrame are replaced with other values dynamically. This differs from updating with .loc or .iloc which require you to specify a location to update with some value.