np.argwhere

np.argwhere函数是NumPy库中的一个函数，用于查找数组中满足某个条件的元素的索引。它返回一个包含满足条件的元素索引的数组。具体来说，np.argwhere(condition)会返回一个二维数组，其中每一行是满足条件的元素的索引。条件可以是任何能够返回布尔值的表达式或函数。

例如，假设有一个一维数组arr，我们想要找到数组中所有大于5的元素的索引，可以使用np.argwhere(arr > 5)。

请注意，np.argwhere返回的是多维数组的索引，所以结果是一个二维数组。如果想要得到一维数组形式的索引，可以使用np.squeeze()函数进行压缩。

希望对你有所帮助！如果还有其他问题，请随时提问。