Come posso trovare le posizioni di un oggetto in un elenco di elenchi Python?

Question

Voglio trovare la posizione (s) di un elemento specifico in un elenco di elenchi. Dovrebbe restituire un elenco di tuple, in cui ogni tupla rappresenta gli indici per un'istanza specifica dell'elemento. Per esempio:

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

Questo è un buon punto di partenza. Presumibilmente si dispone di una sorta di classe come la

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

che in questo caso avrebbe lasciato che dici

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

Si noti che in entrambi i casi si otterrà un'eccezione se il tuo obiettivo non è in ogni uno dei tuoi sottolisti. La domanda non specifica se questo è il comportamento desiderato.

Answer 2

Sembra che si desideri, per un elenco di sottoliste e un determinato articolo, restituire un elenco di coppie in cui ciascuna coppia è (l'indice della sottolista, l'indice dell'elemento all'interno della sottolista). È possibile farlo usando list comprehensions e Python di costruito in funzione di enumerate(): risposta vedere @ di scarabocchio sopra/sotto:

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

Modifica.

Answer 3

Se non si desidera un'eccezione se l'elemento non è presente nell'elenco, provare questo. Anche come generatore perché sono fantastici e versatili.

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

Se si desidera qualcosa che saranno entrambi

• trovare i duplicati e
• maniglia liste annidate (elenchi delle liste di liste di ...)

si può fare qualcosa come il seguente:

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

Testing questo:

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

Qualche tempo fa ho scritto una libreria per Python per fare la lista di corrispondenza che si adatterebbe il disegno di legge abbastanza bene. Ha usato i token?, + E * come caratteri jolly, dove? significa un singolo atomo, + è uno-o-più avido, e * è uno-o-più avido. Per esempio:

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

scaricarlo qui: http://www.artfulcode.net/wp-content/uploads/2008/12/matching.zip

Answer 6

Ecco una versione senza provare ..tranne, restituendo un iteratore e che per

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

rendimenti

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

Il più strainghtforward e probabilmente il modo più lento da eseguire sarebbe:

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

Ecco un altro metodo dritto in avanti quello non usa i generatori.

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]