如何在Python列表中找到项目的位置?

时间:2009-05-12 14:18:37

标签: python

我想在列表列表中找到特定项目的位置。它应该返回一个元组列表,其中每个元组代表该项的特定实例的索引。例如:

list = [['1', '2', '4', '6'], ['7', '0', '1', '4']]
getPosition('1')  #returns [(0, 0), (1, 2)]
and getPosition('7') #returns [(1,0)]

9 个答案:

答案 0 :(得分:7)

如果你想要的东西

  • 找到重复项和
  • 处理嵌套列表(列表列表......列表)

您可以执行以下操作:

def get_positions(xs, item):
    if isinstance(xs, list):
        for i, it in enumerate(xs):
            for pos in get_positions(it, item):
                yield (i,) + pos
    elif xs == item:
        yield ()

测试:

>>> xs = [['1', '2', '4', '6'],
...       ['7', '0', '1', '4'],
...       [ [ '0', '1', '1'], ['1']]
...       ]
>>> print list(get_positions(xs, '1'))
[(0, 0), (1, 2), (2, 0, 1), (2, 0, 2), (2, 1, 0)]

答案 1 :(得分:5)

对于子列表和给定项目的列表,您看起来想要返回每对对的列表列表(子列表的索引,子列表中项目的索引)。您可以使用list comprehensions和Python内置的enumerate()函数:

来实现
def getPosition(list, item):
    return [(i, sublist.index(item)) for i, sublist in enumerate(list)]

修改:请参阅上面/下方的@ scribble回答。

答案 2 :(得分:3)

def getPosition(list, item):
    return [(i, sublist.index(item)) for i, sublist in enumerate(list) 
                                                      if item in sublist]

答案 3 :(得分:2)

def get_positions(xs, target):
    return [(i,e.index(target)) for i,e in enumerate(xs)]

这是一个很好的起点。大概你有某种类,比如

class SomeClass:
    def __init__(self):
        self.xs = [['1','2','4','6'], ['7','0','1','4']]

    def get_positions(self, target):
        return [(i,e.index(target)) for i,e in enumerate(self.xs)]

在这种情况下会让你说

model = SomeClass()
model.get_position(1)    # returns [(0,0), (1,2)]

请注意,在这两种情况下,如果目标不在每个子列表中,您将收到异常。问题没有说明这是否是所期望的行为。

答案 4 :(得分:2)

如果您不希望例外,如果该项目不在列表中,请尝试此操作。也作为发电机,因为它们很酷且功能多样。

xs = [['1', '2', '4', '6'], ['7', '0', '1', '4']]
def get_positions(xs, item):
    for i, xt in enumerate( xs ):
        try: # trying beats checking
            yield (i, xt.index(item))
        except ValueError: 
            pass

print list(get_positions(xs, '1'))
print list(get_positions(xs, '6'))

# Edit for fun: The one-line version, without try:

get_positions2 = lambda xs,item: ((i,xt.index(item)) for  i, xt in enumerate(xs) if item in xt)

print list(get_positions2(xs, '1'))
print list(get_positions2(xs, '6'))

答案 5 :(得分:0)

前段时间我为python编写了一个库,可以进行列表匹配,非常适合该法案。它使用令牌?,+和*作为通配符,在哪里?表示单个原子,+是非贪婪的一个或多个,*是贪婪的一个或多个。例如:

from matching import match

match(['?', 2, 3, '*'], [1, 2, 3, 4, 5])
=> [1, [4, 5]]

match([1, 2, 3], [1, 2, 4])
=> MatchError: broken at 4

match([1, [2, 3, '*']], [1, [2, 3, 4]])
=> [[4]]

match([1, [2, 3, '*']], [1, [2, 3, 4]], True)
=> [1, 2, 3, [4]]

在此处下载:http://www.artfulcode.net/wp-content/uploads/2008/12/matching.zip

答案 6 :(得分:0)

这是一个没有try..except的版本,返回一个迭代器和

[['1', '1', '1', '1'], ['7', '0', '4']]

返回

[(0, 0), (0, 1), (0, 2), (0, 3)] 


def getPosition1(l, val):
  for row_nb, r in enumerate(l):
      for col_nb in (x for x in xrange(len(r)) if r[x] == val):
         yield row_nb, col_nb

答案 7 :(得分:0)

最直接也可能是最慢的方法是:

    >>> value = '1'
    >>> l = [['1', '2', '3', '4'], ['3', '4', '5', '1']]
    >>> m = []
    >>> for i in range(len(l)):
    ...  for j in range(len(l[i])):
    ...   if l[i][j] == value:
    ...    m.append((i,j))
    ...
    >>> m
    [(0, 0), (1, 3)]

答案 8 :(得分:0)

这是另一种不使用生成器的直接方法。

def getPosition(lists,item):
    positions = []
    for i,li in enumerate(lists):
        j = -1
        try:
            while True:
                j = li.index(item,j+1)
                positions.append((i,j))
        except ValueError:
            pass
    return positions

l = [['1', '2', '4', '6'], ['7', '0', '1', '4']]
getPosition(l,'1')  #returns [(0, 0), (1, 2)]
getPosition(l,'9') # returns []

l = [['1', '1', '1', '1'], ['7', '0', '1', '4']]
getPosition(l,'1')  #returns [(0, 0), (0, 1), (0,2), (0,3), (1,2)]