如何在Python中基于逻辑数组设置Numpy数组指定行的值？

问题背景 #

你需要将数据数组中对应逻辑数组值为1的行全部设为0，对应MATLAB中的data[ind,:] = 0操作，但NumPy对整数数组的索引逻辑与MATLAB不同，需要做简单调整。

实现方法 #

NumPy中，整数数组[1,0,1]会被当作位置索引（即提取第1、0、1行），而非逻辑筛选。要实现MATLAB的逻辑索引效果，需将整数数组转换为布尔数组：

1. 先定义数组：

import numpy as np

ind = np.array([1, 0, 1])
data = np.array([[1,2,3],
                 [4,5,6],
                 [7,8,9]])

2. 执行赋值操作（两种等价方式）：

• 方式一：将整数数组转为布尔类型

data[ind.astype(bool), :] = 0

• 方式二：通过条件判断生成布尔数组

data[ind == 1, :] = 0

3. 验证结果：

print(data)
# 输出：
# [[0 0 0]
#  [4 5 6]
#  [0 0 0]]

关键说明 #

MATLAB会自动将1/0视为逻辑值true/false，但NumPy需要显式区分位置索引和逻辑索引。通过将整数数组转换为布尔数组，就能实现和MATLAB一致的逻辑筛选赋值效果。

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