Eliminar elementos de una lista anidada Python

Question

Estoy tratando de eliminar elementos de una lista anidada en Python. Tengo una lista anidada de la siguiente manera:

families = [[0, 1, 2],[0, 1, 2, 3],[0, 1, 2, 3, 4],[1, 2, 3, 4, 5],[2, 3, 4, 5, 6]] 

Quiero eliminar las entradas de cada sublista que coorespond a la posición de indexado de la lista secundaria en la lista maestra. Entonces, por ejemplo, necesito eliminar 0 de la primera sublista, 1 de la segunda sublista, etc. Estoy tratando de usar una lista de aprehensión, haz esto. Esto es lo que he intentado:

familiesNew = [ [ families[i][j] for j in families[i] if i !=j ] for i in range(len(families)) ] 

Esto funciona para range(len(families)) hasta 3, sin embargo, más allá de eso me sale IndexError: list index out of range. No estoy seguro por qué. ¿Alguien puede darme una idea de cómo hacer esto? Preferiblemente un trazador de líneas (lista de comprensión).

Gracias.

Answer 1

Casi lo tienes bien. Basta con sustituir families[i][j] con j y funciona:

>>> [ [ j for j in families[i] if i !=j ] for i in range(len(families)) ] 
[[1, 2], [0, 2, 3], [0, 1, 3, 4], [1, 2, 4, 5], [2, 3, 5, 6]] 

Se puede escribir un poco más limpia usando la función enumerate:

>>> [[f for f in family if f != i] for i, family in enumerate(families)] 
[[1, 2], [0, 2, 3], [0, 1, 3, 4], [1, 2, 4, 5], [2, 3, 5, 6]] 

O incluso utilizando remove si no te importa el cambio de la lista original:

>>> for i, family in enumerate(families): family.remove(i) 

Answer 2

¿Hace esto que usted quiere?

familiesNew=[ filter(lambda x:x!=i,j) for i,j in enumerate(families) ] 

EDITAR

Tenga en cuenta también, la razón por la suya fallidos se debe a que en el tercer elemento de la lista externa ([1, 2, 3, 4, 5]) que está tratando de conseguir el quinto elemento en su bucle (= for j in families[i] = for j in [1,2,3,4,5]), pero las familias [i] tiene una longitud de 5, lo que significa que el índice es mayor 4. lo siento si esa explicación es un poco confuso ... tal vez el siguiente le ayudará a aclararlo un poco:

families = [[0, 1, 2],[0, 1, 2, 3],[0, 1, 2, 3, 4],[1, 2, 3, 4, 5],[2, 3, 4, 5, 6]] 

def f(i,j): 
    print i,j,families[i] 
    return families[i][j] 
#THIS DOES NOT WORK -- but it will tell you where it failed. 
familiesNew = [ [ f(i,j) for j in families[i] if i !=j ] for i in range(len(families)) ] 

Answer 3

Pregunta editada, eliminando la respuesta que estaba resolviendo el problema incorrecto. Además, añadió respuesta adicional por @Ashwini:

Para efectos de comparación:

root# python -m timeit 'families = [[0, 1, 2],[0, 1, 2, 3],[0, 1, 2, 3, 4],[1, 2, 3, 4, 5],[2, 3, 4, 5, 6]]' '[x.remove(ind) for ind,x in enumerate(families) ]' 
100000 loops, best of 3: 3.42 usec per loop  

root# python -m timeit -s 'families = [[0, 1, 2],[0, 1, 2, 3],[0, 1, 2, 3, 4],[1, 2, 3, 4, 5],[2, 3, 4, 5, 6]]' '[[f for f in family if f != i] for i, family in enumerate(families)]' 
100000 loops, best of 3: 4.87 usec per loop 

root# python -m timeit -s 'families = [[0, 1, 2],[0, 1, 2, 3],[0, 1, 2, 3, 4],[1, 2, 3, 4, 5],[2, 3, 4, 5, 6]]' '[ filter(lambda x:x!=i,j) for i,j in enumerate(families) ]' 
100000 loops, best of 3: 7.99 usec per loop 

Estos son micro-segundos, así que creo que lo que quiere hacer está bien a menos que usted va a estar haciendo esto un lote veces.

Answer 4

Si desea modificar la lista original, entonces intente esto:

>>>[x.remove(ind) for ind,x in enumerate(families) ] 
>>>families 
[[1, 2], [0, 2, 3], [0, 1, 3, 4], [1, 2, 4, 5], [2, 3, 5, 6]]