Python中满足条件时替换矩阵行元素的方法（含末行替换示例）

嘿，这个需求很典型，我给你两种实用方案，覆盖不同场景：

方法一：原生Python列表实现 #

如果你的矩阵是普通的二维列表，不用第三方库也能轻松搞定。核心思路是定位到目标行，然后遍历该行的每个元素，判断条件后替换：

# 示例矩阵
matrix = [
    [1, 4, 6],
    [2, 5, 7],
    [3, 1, 2]  # 最后一行
]

# 获取最后一行（索引为-1，Python里负数索引从末尾开始数）
last_row = matrix[-1]

# 遍历替换：小于3的元素设为0
for i in range(len(last_row)):
    if last_row[i] < 3:
        last_row[i] = 0

print(matrix)

运行后输出：

[
    [1, 4, 6],
    [2, 5, 7],
    [3, 0, 0]
]

方法二：使用numpy高效实现 #

如果你经常处理矩阵、数值计算，numpy绝对是最佳选择，它的布尔索引能让代码简洁又高效：

import numpy as np

# 把列表转成numpy数组
arr = np.array([
    [1, 4, 6],
    [2, 5, 7],
    [3, 1, 2]
])

# 直接用条件索引替换：最后一行中小于3的元素设为0
arr[-1, arr[-1] < 3] = 0

print(arr)

运行后输出：

[[1 4 6]
 [2 5 7]
 [3 0 0]]

这里解释下关键代码arr[-1, arr[-1] < 3] = 0：

• arr[-1] 获取数组的最后一行
• arr[-1] < 3 生成一个布尔数组，对应位置元素小于3时为True
• 用这个布尔数组作为索引，选中最后一行中需要替换的元素，直接赋值为0

这种写法比原生列表快得多，尤其是当矩阵规模很大的时候。

内容的提问来源于stack exchange，提问作者Alberto Alvarez