python 在矩阵中查询值

我们可以使用 NumPy 库来进行矩阵操作。假设我们有一个矩阵 `matrix`，我们希望查询其中值为 `value` 的元素的位置，可以使用如下代码：

import numpy as np

# 假设这是我们要查询的矩阵
matrix = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

# 要查询的值
value = 5

# 使用 np.where() 函数查询值等于 value 的元素的位置
indices = np.where(matrix == value)
print(indices)

输出结果为：

(array([1]), array([1]))

其中，`indices` 是一个元组，第一个元素是行的索引，第二个元素是列的索引。在这个例子中，值为 `5` 的元素在第二行第二列，所以输出的 `indices` 为 `(array([1]), array([1]))`。