User np.argwhere()与np.where()

`np.argwhere()`和`np.where()`都是NumPy库中的函数，用于在数组中查找满足条件的元素的索引。

`np.where()`函数返回满足条件的元素的索引，而`np.argwhere()`函数返回满足条件的元素的坐标，即每个元素的索引值组成的元组。

以下是两个函数的使用示例：

import numpy as np

# 创建一个二维数组
arr = np.array([[0, 1, 2], [3, 4, 5], [6, 7, 8]])

# 使用np.where()函数查找数组中大于3的元素的索引
indices = np.where(arr > 3)
print(indices)  # 输出：(array([1, 1, 2, 2, 2]), array([1, 2, 0, 1, 2]))

# 使用np.argwhere()函数查找数组中大于3的元素的坐标
coordinates = np.argwhere(arr > 3)
print(coordinates)  # 输出：[[1 1][1 2][2 0][2 1][2 2]]