python查找矩阵中的元素索引

可以使用numpy库的argwhere函数来查找矩阵中元素的索引。具体实现如下：

import numpy as np

# 定义一个矩阵
matrix = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

# 查找元素5的索引
indices = np.argwhere(matrix == 5)

print(indices)
# 输出结果为：[[1 1]]

以上代码中，argwhere函数会返回所有满足条件的元素的索引值，这里返回的是元素值为5的索引。如果有多个满足条件的元素，那么返回的索引将会是一个二维数组，每一行代表一个元素的索引。