在python中找到一个numpy数组中元素的位置[重复]

Question

我想问一个关于在 Python 的 numpy 包中查找数组中元素位置的问题。

我正在使用适用于 Python 3 的 Jupyter Notebook，下面的代码如下所示：

concentration_list = array([172.95, 173.97, 208.95])

我想编写一段代码，它能够返回数组中元素的位置。

为此，我想用172.95来演示。

最初，我尝试使用.index()，在括号内传入172.95，但这不起作用，因为numpy无法识别.index()方法-

concentration_position = concentration_list.index(172.95)

AttributeError: 'numpy.ndarray' object has no attribute 'index'

当我访问该站点时，Sci.py 文档没有提到任何关于这种方法可用的信息。

是否有任何可用的功能（我可能没有发现）来解决问题？

Answer 1

您可以通过numpy 库中的where 函数

import numpy as np

concentration_list = np.array([172.95, 173.97, 208.95])
number = 172.95

print(np.where(concentration_list == number)[0])


Output : [0]

Answer 2

为此目的使用np.where(...)，例如

import numpy as np

concentration_list = np.array([172.95, 173.97, 208.95])

index=np.ravel(np.asarray(concentration_list==172.95).nonzero())

print(index)

#outputs (array of all indexes matching the condition):

>> [0]