¿Cómo puedo encontrar las ubicaciones de un elemento en una lista de listas de Python?

Question

Quiero encontrar la ubicación (es) de un elemento específico en una lista de listas. Debería devolver una lista de tuplas, donde cada tupla representa los índices de una instancia específica del artículo. Por ejemplo:

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

Answer 1

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

Ese es un buen punto de partida. Es de suponer que usted tiene algún tipo de clase como

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)] 

que en este caso sería dejar que usted dice

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

Nótese que en ambos casos obtendrá una excepción si su objetivo no es en todos una de tus sublistas La pregunta no especifica si este es el comportamiento deseado.

Answer 2

Parece que desea, para una lista de sublistas y un elemento determinado, devolver una lista de pares donde está cada par (el índice de la sublista, el índice del artículo dentro de la sublista). Puede hacer que el uso de list comprehensions y Python de enumerate() construida en función de: la respuesta @ Ver garabato de arriba/abajo:

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

Editar.

Answer 3

Si no desea una excepción si el artículo no está en la lista, intente esto. También como generador porque son geniales y versátiles.

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')) 

Answer 4

Si quieres algo que a la vez

• encontrar duplicados y
• mango listas anidadas (listas de listas de listas de ...)

se puede hacer algo como el siguiente:

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() 

Prueba esto:

>>> 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)] 

Answer 5

Hace un tiempo escribí una biblioteca de Python para hacer lista de coincidencias que se ajuste a la factura bastante bien. Usó los tokens?, + Y * como comodines, ¿dónde? significa un átomo individual, + es uno o más no codicioso, y * es codicioso uno o más. Por ejemplo:

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]] 

Descargar aquí: http://www.artfulcode.net/wp-content/uploads/2008/12/matching.zip

Answer 6

Aquí es una versión sin tratar ..excepto, volviendo un iterador y que por

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

vuelve

[(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 

Answer 7

El más strainghtforward y probablemente la manera más lenta que hacer sería:

>>> 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)] 

Answer 8

Aquí hay otro método directo y que no usa generadores

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)] 

Answer 9

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