当只查找2D数组中的一个项目时,如何在2D数组中查找内容?

时间:2015-03-10 17:31:47

标签: python arrays python-3.x multidimensional-array

例如,如果我有

array=[["A",1],["B",2],["C",1]]

通过寻找" A"有什么方法可以找到[" A",1]?我试图在数组中的第一个东西是唯一的情况下使用它,所以看第二个没有意义,我也无法知道第二个变量是什么

2 个答案:

答案 0 :(得分:2)

迭代外部列表中存在的项目,并检查内部列表的第一个元素是否满足特定条件。

>>> a=[["A",1],["B",2],["C",1]]
>>> next(i for i in a if i[0] == 'A')
['A', 1]
>>> [i for i in a if i[0] == 'A']
[['A', 1]]

答案 1 :(得分:0)

如果您控制数据类型,取决于您使用此对象的其他方式,字典可能是更好的选择:

而不是

array=[["A",1],["B",2],["C",1]]

使用

d={"A":1, "B":2, "C":1}

然后,您只需使用

即可访问与"A"相关联的元素
>> d["A"]

1

如果您想将列表转换为字典:

d = dict(array)