python 如果一个dataframe中一个值为0,那么就将该值所在的行和列都变为0

python 如果一个dataframe中一个值为0,那么就将该值所在的行和列都变为0,应该怎么写?
我的代码哪里出错了

img

应该复制数据到新的dataframe中,遍历这个克隆的原始数据,不能直接操作df1,要不会出现0后更改行列值会影响后续未遍历到的数据导致设置出错

import pandas as pd

def setDataFrameZeros(df):
    dfclone=df.copy()#复制数据
    for i in range(dfclone.shape[0]):
        for a in range(dfclone.shape[1]):
            if dfclone.iloc[i,a]==0:
                df.iloc[i,:]=0
                df.iloc[:,a]=0
    return df


df1=pd.DataFrame({'c1':[1,4,7],
                  'c2':[2,0,8],
                  'c3':[3,6,9]
    })

df2=pd.DataFrame({'c1':[1,0,7],
                  'c2':[0,0,0],
                  'c3':[3,0,9]
    })

assert df2.equals(setDataFrameZeros(df1))
print(df1)

有问题,比如第一次满足if条件,第一行、第一列都被修改为0 了,影响了后续遍历的结果。

您好,我是有问必答小助手,您的问题已经有小伙伴帮您解答,感谢您对有问必答的支持与关注!
PS:问答VIP年卡 【限时加赠:IT技术图书免费领】,了解详情>>> https://vip.csdn.net/askvip?utm_source=1146287632