如何获取一个布尔矩阵中True所对应的行列索引
时间: 2024-03-10 13:51:25 浏览: 55
matlab布尔矩阵产生代码-MultiViewMoSeg:MultiViewMoSeg
可以使用numpy库中的函数`np.argwhere()`来获取一个布尔矩阵中True所对应的行列索引。该函数返回一个二维数组,其中每个子数组表示一个True值的索引,子数组的第一个元素是行索引,第二个元素是列索引。例如,对于以下布尔矩阵:
```
[[ True False False]
[False True False]
[False False True]]
```
可以使用`np.argwhere()`函数来获取True值所对应的索引:
```python
import numpy as np
bool_matrix = np.array([[True, False, False], [False, True, False], [False, False, True]])
true_indexes = np.argwhere(bool_matrix)
print(true_indexes)
```
输出结果为:
```
[[0 0]
[1 1]
[2 2]]
```
其中,第一行表示第一个True值所在的行列索引,第二行表示第二个True值所在的行列索引,以此类推。
阅读全文