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)
有问题,比如第一次满足if条件,第一行、第一列都被修改为0 了,影响了后续遍历的结果。
您好,我是有问必答小助手,您的问题已经有小伙伴帮您解答,感谢您对有问必答的支持与关注!