如何基于多条件将NumPy数组指定索引的行最大值置0？

你的代码存在两个核心问题：

1. np.amax(arr[:,1])取的是整个第二列的最大值，而非每行的最大值，不符合需求；
2. 逻辑判断的括号位置错误，导致布尔运算优先级出错。

以下是修正后的实现：

import numpy as np

arr = np.array([[1, 5, 3, 3], 
                [5, 6, 9, 3], 
                [2, 9, 3, 3], 
                [1, 4, 2, 2]])

# 1. 计算每行的最大值
row_max = np.amax(arr, axis=1)
# 2. 生成需要替换的条件：索引1处是该行最大值，且值小于7
replace_mask = (arr[:, 1] == row_max) & (arr[:, 1] < 7)
# 3. 复制原数组避免修改原数据
output = arr.copy()
# 4. 对符合条件的行的索引1位置赋值为0
output[replace_mask, 1] = 0

print(output)

运行后输出：

[[1 0 3 3]
 [5 6 9 3]
 [2 0 3 3]
 [1 0 2 2]]

关键步骤说明 #

• np.amax(arr, axis=1)：指定axis=1按行计算最大值，得到形状为(4,)的数组，对应每行的最大值；
• replace_mask：组合两个条件生成布尔数组，标记哪些行需要修改索引1的位置；
• 使用arr.copy()：避免直接修改原数组，保证数据安全性；
• output[replace_mask, 1] = 0：利用布尔索引精准定位需要替换的元素位置并赋值。

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