python |从二维数组中提取元素

Question

由于我对 Python 还很陌生，所以我什至不确定如何表达我的问题。我想要完成的基本概念是能够在二维数组中搜索某些内容并检索正确的值以及与该值关联的值(抱歉我的错误解释)

例如

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

如果用户想要找到 2，我希望程序检索 [2,x,d]，如果可能的话，将其放入普通 (1D)大批。同样，如果用户搜索 3，程序应该检索 [3,c,f]。

预先感谢您(如果可能的话，我想要一个不涉及 numpy 的解决方案)

Answer 1

你可以做一个简单的 for 循环，并使用内置的 in 语句:

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