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

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

应该复制数据到新的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)