Python: Wybierz podzbiór z listy na podstawie indeksu ustawić

Question

Mam kilka list posiadające wszystkie taką samą liczbę wpisów (każdy określające właściwość Object):

property_a = [545., 656., 5.4, 33.] 
property_b = [ 1.2, 1.3, 2.3, 0.3] 
... 

i listy z flagami tej samej długości

good_objects = [True, False, False, True] 

(co może być łatwo podstawiona przez równoważną liście indeksów:

good_indices = [0, 3] 

co jest najprostszym wag ay do generowania nowych list property_asel, property_bsel, ... które zawierają tylko wartości wskazane przez wpisy lub indeksy?

property_asel = [545., 33.] 
property_bsel = [ 1.2, 0.3] 

Answer 1

może po prostu użyć list comprehension:

property_asel = [val for is_good, val in zip(good_objects, property_a) if is_good] 

lub

property_asel = [property_a[i] for i in good_indices] 

Ten ostatni jest szybszy, ponieważ istnieje mniej good_indices niż długość property_a, zakładając good_indices są precomputed zamiast generowane w locie.

---

Edycja: Opcja pierwsza odpowiada itertools.compress dostępny od Pythonie 2.7/3.1. Zobacz odpowiedź: @Gary Kerr.

property_asel = list(itertools.compress(good_objects, property_a)) 

Answer 2

Widzę 2 opcje.

1. Korzystanie numpy:

   property_a = numpy.array([545., 656., 5.4, 33.]) 
   property_b = numpy.array([ 1.2, 1.3, 2.3, 0.3]) 
   good_objects = [True, False, False, True] 
   good_indices = [0, 3] 
   property_asel = property_a[good_objects] 
   property_bsel = property_b[good_indices] 
   

2. Korzystanie z listy ze zrozumieniem i zip go:

   property_a = [545., 656., 5.4, 33.] 
   property_b = [ 1.2, 1.3, 2.3, 0.3] 
   good_objects = [True, False, False, True] 
   good_indices = [0, 3] 
   property_asel = [x for x, y in zip(property_a, good_objects) if y] 
   property_bsel = [property_b[i] for i in good_indices] 
   

Answer 3

korzystać z wbudowanych funkcji zip

property_asel = [a for (a, truth) in zip(property_a, good_objects) if truth] 

---

EDIT

Po prostu patrząc na nowych funkcji 2.7. W module itertools jest teraz funkcja podobna do powyższego kodu.

http://docs.python.org/library/itertools.html#itertools.compress

itertools.compress('ABCDEF', [1,0,1,0,1,1]) => 
    A, C, E, F 

Answer 4

Zakładając masz tylko listę elementów i listę prawdziwych/wymaganych wskaźników, powinno to być najszybszy:

property_asel = [ property_a[index] for index in good_indices ] 

Oznacza to wybór nieruchomości będzie nie tylko tyle rund, ile jest prawdziwych/wymaganych wskaźników.Jeśli masz dużo wykazów nieruchomości, które następują z zasadami pojedynczych znaczników (prawda/fałsz) listę można utworzyć listę indeksów przy użyciu tych samych zasad lista ze zrozumieniem:

good_indices = [ index for index, item in enumerate(good_objects) if item ] 

This iteracji każdej pozycji good_objects (podczas zapamiętywania indeksu z wyliczeniem) i zwraca tylko indeksy, w których pozycja jest prawdziwa.

---

Dla nikogo nie otrzymuję listy ze zrozumieniem, tutaj jest wersja proza ​​angielski z kodem pogrubione:

lista indeks dla każda grupa indeksu, poz że istnieje wwyliczenie z dobrych obiektów, jeśli (gdzie) pozycja True

Answer 5

Języki Matlab i Scilab oferują prostszą i bardziej elegancką składnię niż w pythonie na pytanie, które zadajesz, więc myślę, że najlepiej jest naśladować Matlab/Scilab przy użyciu pakietu Numpy w Pythonie. W ten sposób rozwiązanie problemu jest bardzo zwięzły i elegancki:

from numpy import * 
property_a = array([545., 656., 5.4, 33.]) 
property_b = array([ 1.2, 1.3, 2.3, 0.3]) 
good_objects = [True, False, False, True] 
good_indices = [0, 3] 
property_asel = property_a[good_objects] 
property_bsel = property_b[good_indices] 

Numpy próbuje naśladować Matlab/Scilab ale chodzi w cenie: trzeba zadeklarować każdej listy ze słowem kluczowym „tablicy”, coś, co przeładuje twój skrypt (ten problem nie istnieje w Matlab/Scilab). Zauważ, że to rozwiązanie jest ograniczone do tablic liczb, co ma miejsce w twoim przykładzie.