基於特定列的值對 numpy 矩陣進行分組，僅在給定數組的行索引上

Question

我有一個由二進制值組成的 numpy 矩陣。 我也有一個行索引列表。 現在我必須從特定列的值為 1 的矩陣中獲取索引，並且索引必須包含在行索引列表中。 什么是這樣做的有效方法？ 我目前正在做：

result = [index for index in np.where(dataset[:, col] == 1)[0] if index in indices]

Answer 1

也許循環遍歷行索引列表會更快，因為您不需要遍歷矩陣的所有行：

result = [index for index in indices if dataset[index, col] == 1]