Python中DataFrame行和为0时如何修改对应位置对角线元素的值

原代码问题分析 #

• 运算符使用错误：& 是位运算符，此处做逻辑判断应使用逻辑运算符and；即使要使用&，也需要给两边的判断条件分别包裹括号，否则会因运算符优先级问题导致计算逻辑完全错误。
• 求和结果类型不匹配：df.iloc[[i]].sum(axis = 1) 返回的是Series类型对象，无法直接用于if的标量判断场景，需要提取为数值标量才能做比较。
• 存在冗余判断：你循环中i和j是同步遍历的，天然相等，不需要额外增加i == j的判断逻辑。

修复后的循环实现 #

如果要沿用你原来的循环写法，修改后可正常运行的代码如下：

for i in range(len(df)):
    if df.iloc[i].sum() == 0:
        df.loc[i, i] = 1

更高效的向量化实现 #

不需要显式写循环，Pandas向量化操作性能更高，尤其适合数据量较大的场景：

# 计算所有行的行和
row_sum = df.sum(axis=1)
# 筛选行和为0的索引
zero_row_idx = row_sum[row_sum == 0].index
# 批量修改对应对角线元素
df.loc[zero_row_idx, zero_row_idx] = 1

如果你的DataFrame行、列索引不是默认的0起始整数序列，用位置索引修改更稳妥：

row_sum = df.sum(axis=1)
zero_row_pos = row_sum[row_sum == 0].index.to_list()
df.iloc[zero_row_pos, zero_row_pos] = 1

内容的提问来源于stack exchange，提问作者Claire Wang