Python | 从2D数组中提取元素

Question

由于我是python的新手，我什至不知道该如何表达我的问题。 我要完成的工作的基本概念是能够在2D数组中搜索某些内容并检索正确的值以及与该值关联的值（对不起，我的解释不好）

例如

array=[[1,a,b],[2,x,d],[3,c,f]]

如果用户要查找2 ，我希望程序检索[2,x,d]并在可能的情况下将其放入常规（1D）数组中。 同样，如果用户搜索3 ，则程序应检索[3,c,f] 。

在此先感谢您（如果可能，我需要一个不涉及numpy的解决方案）

Answer 1

也许是这样的吗？

def search(arr2d, value):
     for row in arr2d:
         if row[0] == value:
             return row

Answer 2

您可以执行一个简单的for循环，并使用内置的in语句：

def retrieve_sub_array(element):
 for sub_array in array:
  if element in sub_array:
   return sub_array

Answer 3

尝试类似的东西：

def find(value, array):
    for l in array:
        if l[0]==value:
            return l

或者，如果您想了解更多信息：

array[list(zip(*array))[0].index(value)]

Answer 4

如果适合您的问题，可以使用字典：

>>> dict={1:['a','b'],2:['x','d'],3:['c','f']}
>>> dict[2]
['x', 'd']

然后，在每个列表中线性搜索正确的索引会更有效。