python - 如何在数组中查找数组的索引

Question

我已经按照下面的方式创建了一个数组；代表 3 对坐标。我的问题是我似乎无法在数组中找到一对特定坐标的索引。

import numpy as np

R = np.random.uniform(size=(3,2))

R

Out[5]: 
array([[ 0.57150157,  0.46611662],
   [ 0.37897719,  0.77653461],
   [ 0.73994281,  0.7816987 ]])

R.index([ 0.57150157,  0.46611662])

返回以下内容:

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

我尝试这样做的原因是我可以在 for 循环中使用坐标对的索引扩展列表。

例如

v = []
for A in R:
v.append(R.index(A)) 

我只是不确定为什么索引函数不起作用，而且似乎找不到解决方法。

我是编程新手，所以如果这看起来像是胡说八道，请原谅。

Answer 1

index() 是 list 类型的方法，而不是 numpy.array 类型的方法。尝试:

R.tolist().index(x)

例如，x 是 R 的第三个条目。这首先将您的数组转换为 list，然后您可以使用 index ;)