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)
读取两次,然后遍历df2,如果满足条件,通过修改df1中的内容,最后输出df1