python - 如何在 Python 中获取数组中列表的索引？

Question

我有一个列表a=[1,2,3,4]和一个数组arr=np.array([[1,1,1,1],[1,2,3,4],[2,2,2,2])。我想比较a并arr获得ain的索引arr。我怎样才能做到这一点？我希望以整数形式获得输出（在本例中为 1），因为我需要索引才能继续其他进程。提前致谢！

标签： pythonarrayslist

Answer 1

获取索引的一种方法是将 numpy 数组转换为 python 嵌套列表，然后在转换后的列表上使用 index method()

（顺便说一下，您的声明中缺少一个结束的“]” arr）

import numpy as np
a = [1, 2, 3, 4]
arr=np.array([[1,1,1,1],[1,2,3,4],[2,2,2,2]])
lst = arr.tolist()
idx = lst.index(a)