从np数组找需要值并返回数组

你可以使用numpy库中的函数来实现从数组中找到所需值并返回数组，具体可以使用以下代码实现：

import numpy as np

# 创建一个数组
arr = np.array([1, 2, 3, 4, 5])

# 找到需要的值并返回数组
result = np.where(arr == 3)

print(result)

在上述代码中，我们使用`np.where()`函数来找到值为3的元素在数组中的位置，并将这些位置信息存储在一个元组中返回，结果为`(array([2]),)`。如果需要返回对应的元素值，可以使用`arr[result]`来获取。

相关问题

np 二维数组取最大值的数组

要获取二维数组中最大值所在的位置，可以使用NumPy库中的argmax函数。该函数返回最大值所在的索引，可以通过该索引获取最大值所在的位置。

例如，假设有一个二维数组arr：

import numpy as np

arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

要获取该数组中最大值所在的位置，可以使用以下代码：

max_index = np.argmax(arr)

该函数会返回一个整数，表示最大值在一维展开后的索引。为了获取该索引在二维数组中的位置，可以使用以下代码：

max_position = np.unravel_index(max_index, arr.shape)

该函数会返回一个元组，表示最大值在二维数组中的位置。例如，在上面的例子中，该函数会返回一个元组(2, 2)，表示最大值位于第3行第3列。

希望能对你有所帮助！

np.argwhere返回空数组

np.argwhere返回空数组的原因可能是输入数组中没有满足条件的元素。np.argwhere函数用于查找满足指定条件的元素的索引。如果没有符合条件的元素，则返回一个空数组。

请检查一下你的输入数组和条件，确保输入数组中存在满足条件的元素。另外，还可以尝试使用其他的条件或者调整条件，看看是否能够得到非空的结果。

以下是一个示例代码，演示了如何使用np.argwhere函数：

import numpy as np

arr = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
condition = arr % 2 == 0
indices = np.argwhere(condition)

if indices.size == 0:
    print("Empty array")
else:
    print(indices)

在这个例子中，我们通过对arr数组进行取余操作，得到一个条件数组condition，然后使用np.argwhere函数找到满足条件的元素的索引。如果找到了满足条件的元素，就打印出索引；否则打印"Empty array"。