尝试编写Python/pandas函数来更新等于0的特定列中的行？

import pandas as pd

def update_zero_rows(df, column_name, new_value):
    # 使用条件筛选找到等于0的行
    zero_rows = df[df[column_name] == 0]
    
    # 更新等于0的行的特定列为新的值
    df.loc[zero_rows.index, column_name] = new_value
    
    return df

# 创建一个示例DataFrame
data = {'A': [1, 2, 0, 4, 0],
        'B': [0, 0, 0, 0, 0],
        'C': [0, 2, 3, 4, 5]}
df = pd.DataFrame(data)

# 调用函数更新等于0的'A'列中的行为新值10
updated_df = update_zero_rows(df, 'A', 10)

print(updated_df)

    A  B  C
0   1  0  0
1   2  0  2
2  10  0  3
3   4  0  4
4  10  0  5