在元组或对象列表上使用Python的list index()方法?

时间:2009-06-03 20:01:51

标签: python list tuples reverse-lookup

Python的列表类型有一个index()方法,它接受一个参数并返回匹配参数的列表中第一个项的索引。例如:

>>> some_list = ["apple", "pear", "banana", "grape"]
>>> some_list.index("pear")
1
>>> some_list.index("grape")
3

是否有一种优雅(惯用)方式将其扩展到复杂对象列表,如元组?理想情况下,我希望能够做到这样的事情:

>>> tuple_list = [("pineapple", 5), ("cherry", 7), ("kumquat", 3), ("plum", 11)]
>>> some_list.getIndexOfTuple(1, 7)
1
>>> some_list.getIndexOfTuple(0, "kumquat")
2

getIndexOfTuple()只是一个接受子索引和值的假设方法,然后返回该子索引上具有给定值的列表项的索引。我希望

有没有办法实现这样的一般结果,使用列表推导或lambas或类似的“内联”?我想我可以编写自己的类和方法,但如果Python已经有办法,我不想重新发明轮子。

12 个答案:

答案 0 :(得分:60)

这个怎么样?

>>> tuple_list = [("pineapple", 5), ("cherry", 7), ("kumquat", 3), ("plum", 11)]
>>> [x for x, y in enumerate(tuple_list) if y[1] == 7]
[1]
>>> [x for x, y in enumerate(tuple_list) if y[0] == 'kumquat']
[2]

正如评论中指出的那样,这将获得所有匹配。要获得第一个,您可以:

>>> [y[0] for y in tuple_list].index('kumquat')
2

评论中对于所有解决方案之间的速度差异进行了很好的讨论。我可能有点偏颇,但我会亲自坚持一个单行,因为我们谈论的速度相对于创建函数和导入模块来解决这个问题是非常微不足道的,但是如果你打算这么做很多您可能希望查看提供的其他答案的元素,因为它们比我提供的更快。

答案 1 :(得分:26)

一段时间后,这些列表理解会变得混乱。

我喜欢这种Pythonic方法:

from operator import itemgetter

def collect(l, index):
   return map(itemgetter(index), l)

# And now you can write this:
collect(tuple_list,0).index("cherry")   # = 1
collect(tuple_list,1).index("3")        # = 2

如果您需要所有代码都具有超高性能:

# Stops iterating through the list as soon as it finds the value
def getIndexOfTuple(l, index, value):
    for pos,t in enumerate(l):
        if t[index] == value:
            return pos

    # Matches behavior of list.index
    raise ValueError("list.index(x): x not in list")

getIndexOfTuple(tuple_list, 0, "cherry")   # = 1

答案 2 :(得分:9)

一种可能性是使用operator模块中的itemgetter函数:

import operator

f = operator.itemgetter(0)
print map(f, tuple_list).index("cherry") # yields 1

itemgetter的调用返回一个函数,对于传递给它的任何内容,它将执行相当于foo[0]的函数。然后使用map将该函数应用于每个元组,将信息提取到新列表中,然后在其上正常调用index

map(f, tuple_list)

相当于:

[f(tuple_list[0]), f(tuple_list[1]), ...etc]

反过来相当于:

[tuple_list[0][0], tuple_list[1][0], tuple_list[2][0]]

给出:

["pineapple", "cherry", ...etc]

答案 3 :(得分:5)

您可以使用列表推导和index()

来完成此操作
tuple_list = [("pineapple", 5), ("cherry", 7), ("kumquat", 3), ("plum", 11)]
[x[0] for x in tuple_list].index("kumquat")
2
[x[1] for x in tuple_list].index(7)
1

答案 4 :(得分:2)

我会将此作为对Triptych的评论,但由于缺乏评级,我无法发表评论:

使用枚举器方法匹配元组列表中的子索引。 e.g。

li = [(1,2,3,4), (11,22,33,44), (111,222,333,444), ('a','b','c','d'),
        ('aa','bb','cc','dd'), ('aaa','bbb','ccc','ddd')]

# want pos of item having [22,44] in positions 1 and 3:

def getIndexOfTupleWithIndices(li, indices, vals):

    # if index is a tuple of subindices to match against:
    for pos,k in enumerate(li):
        match = True
        for i in indices:
            if k[i] != vals[i]:
                match = False
                break;
        if (match):
            return pos

    # Matches behavior of list.index
    raise ValueError("list.index(x): x not in list")

idx = [1,3]
vals = [22,44]
print getIndexOfTupleWithIndices(li,idx,vals)    # = 1
idx = [0,1]
vals = ['a','b']
print getIndexOfTupleWithIndices(li,idx,vals)    # = 3
idx = [2,1]
vals = ['cc','bb']
print getIndexOfTupleWithIndices(li,idx,vals)    # = 4

答案 5 :(得分:2)

受到this question的启发,我发现这很优雅:

>>> tuple_list = [("pineapple", 5), ("cherry", 7), ("kumquat", 3), ("plum", 11)]
>>> next(i for i, t in enumerate(tuple_list) if t[1] == 7)
1
>>> next(i for i, t in enumerate(tuple_list) if t[0] == "kumquat")
2

答案 6 :(得分:1)

好吧,这可能是vals(j)的错误,更正是:

def getIndex(li,indices,vals):
for pos,k in enumerate(lista):
    match = True
    for i in indices:
        if k[i] != vals[indices.index(i)]:
            match = False
            break
    if(match):
        return pos

答案 7 :(得分:1)

tuple_list = [("pineapple", 5), ("cherry", 7), ("kumquat", 3), ("plum", 11)]

def eachtuple(tupple, pos1, val):
    for e in tupple:
        if e == val:
            return True

for e in tuple_list:
    if eachtuple(e, 1, 7) is True:
        print tuple_list.index(e)

for e in tuple_list:
    if eachtuple(e, 0, "kumquat") is True:
        print tuple_list.index(e)

答案 8 :(得分:0)

z = list(zip(*tuple_list))
z[1][z[0].index('persimon')]

答案 9 :(得分:0)

没有人建议lambdas?

Y尝试这个并且有效。我来这个帖子搜索答案。我没有发现我喜欢,但我感觉到了一种侮辱:P

    l #[['rana', 1, 1], ['pato', 1, 1], ['perro', 1, 1]]

    map(lambda x:x[0], l).index("pato") #1 

编辑以添加示例:

   l=[['rana', 1, 1], ['pato', 2, 1], ['perro', 1, 1], ['pato', 2, 2], ['pato', 2, 2]]

按条件提取所有项目:         filter(lambda x:x [0] ==“pato”,l)#[['pato',2,1],['pato',2,2],['pato',2,2]] < / p>

按条件提取所有项目:

    >>> filter(lambda x:x[1][0]=="pato", enumerate(l))
    [(1, ['pato', 2, 1]), (3, ['pato', 2, 2]), (4, ['pato', 2, 2])]
    >>> map(lambda x:x[1],_)
    [['pato', 2, 1], ['pato', 2, 2], ['pato', 2, 2]]

注意:_变量仅适用于交互式解释器y普通文本文件_需要explicti assign,即_ = filter(lambda x:x [1] [0] ==“pato”,枚举(l))

答案 10 :(得分:0)

Python的list.index(x)返回列表中第一次出现的x的索引。所以我们可以传递列表压缩返回的对象来获取它们的索引。

>>> tuple_list = [("pineapple", 5), ("cherry", 7), ("kumquat", 3), ("plum", 11)]
>>> [tuple_list.index(t) for t in tuple_list if t[1] == 7]
[1]
>>> [tuple_list.index(t) for t in tuple_list if t[0] == 'kumquat']
[2]

使用相同的行,如果有多个匹配的元素,我们也可以获取索引列表。

>>> tuple_list = [("pineapple", 5), ("cherry", 7), ("kumquat", 3), ("plum", 11), ("banana", 7)]
>>> [tuple_list.index(t) for t in tuple_list if t[1] == 7]
[1, 4]

答案 11 :(得分:0)

我想以下并不是做到这一点的最佳方法(速度和优雅感),但是它可能会有所帮助:

from collections import OrderedDict as od
t = [('pineapple', 5), ('cherry', 7), ('kumquat', 3), ('plum', 11)]
list(od(t).keys()).index('kumquat')
2
list(od(t).values()).index(7)
7
# bonus :
od(t)['kumquat']
3

具有2个成员的元组列表可以直接转换为有序dict,数据结构实际上是相同的,因此我们可以即时使用dict方法。