numpy where argwhere

numpy中的where函数和argwhere函数都可以用于根据条件返回数组中的元素索引。其中，where函数返回满足条件的元素的索引，而argwhere函数返回满足条件的元素的坐标。

具体来说，where函数的用法如下：

numpy.where(condition[, x, y])

其中，condition是一个布尔型数组，x和y是可选参数，表示满足条件和不满足条件时返回的值。如果只传入condition，则返回满足条件的元素的索引。

argwhere函数的用法如下：

numpy.argwhere(a)

其中，a是一个数组，返回满足条件的元素的坐标。

举个例子：

import numpy as np

a = np.array([1, 2, 3, 4, 5])
condition = a > 3

# 使用where函数返回满足条件的元素的索引
print(np.where(condition))  # 输出 (array([3, 4]),)

# 使用argwhere函数返回满足条件的元素的坐标
print(np.argwhere(condition))  # 输出 [[3], [4]]