Comprensioni in Python per campionare le tuple da un elenco

Question

Sto cercando di ottenere l'elenco delle tuple a tre elementi dall'elenco [-4, -2, 1, 2, 5, 0] utilizzando le comprensibilità e controllando se soddisfano la condizione sum([] == 0). Il seguente codice funziona. Tuttavia, non c'è dubbio che ci dovrebbe essere un più facile, modo molto più elegante di esprimere queste comprensioni:

[ 
    (i, j, k) for i in [-4, -2, 1, 2, 5, 0] 
       for j in [-4, -2, 1, 2, 5, 0] 
       for k in [-4, -2, 1, 2, 5, 0] if sum([i, j, k]) == 0 
] 

uscita:

[(-4, 2, 2), (-2, 1, 1), (-2, 2, 0), (-2, 0, 2), (1, -2, 1), 
(1, 1, -2), (2, -4, 2), (2, -2, 0), (2, 2, -4), (2, 0, -2), 
(0, -2, 2), (0, 2, -2), (0, 0, 0)] 

La domanda è alla ricerca di un'espressione come (i, j, k) for i, j, k in [-4, -2, 1, 2, 5, 0].

Answer 1

È possibile utilizzare itertools.product per nascondere i cicli nidificati nella comprensione dell'elenco. Utilizzare il parametro repeat per impostare il numero di cicli nel corso della lista (cioè il numero di elementi nella tupla):

valori

>>> import itertools 
>>> lst = [-4, -2, 1, 2, 5, 0] 
>>> [x for x in itertools.product(lst, repeat=3) if sum(x) == 0] 
[(-4, 2, 2), 
(-2, 1, 1), 
(-2, 2, 0), 
(-2, 0, 2), 
(1, -2, 1), 
(1, 1, -2), 
(2, -4, 2), 
(2, -2, 0), 
(2, 2, -4), 
(2, 0, -2), 
(0, -2, 2), 
(0, 2, -2), 
(0, 0, 0)]