ฉันมีหลายรายการที่มีจำนวนรายการเท่ากันทั้งหมด (แต่ละรายการระบุคุณสมบัติของวัตถุ):

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, property_bsel... ซึ่งมีเฉพาะค่าที่ระบุอย่างใดอย่างหนึ่งตามTrueรายการหรือดัชนี?

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

คุณสามารถใช้ความเข้าใจรายการ :

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

หรือ

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

อันหลังเร็วกว่าเนื่องจากมีgood_indicesความยาวน้อยกว่าproperty_aโดยสมมติว่าgood_indicesมีการคำนวณล่วงหน้าแทนที่จะสร้างขึ้นทันที

แก้ไข : ตัวเลือกแรกเทียบเท่ากับที่itertools.compressมีให้ตั้งแต่ Python 2.7 / 3.1 ดูคำตอบของ@Gary Kerr

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

ฉันเห็น 2 ตัวเลือก

1. ใช้ 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. ใช้ความเข้าใจในรายการและซิป:

   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]
   

ใช้ซิปฟังก์ชันในตัว

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

แก้ไข

เพียงแค่ดูคุณสมบัติใหม่ของ 2.7 ขณะนี้มีฟังก์ชันในโมดูล itertools ซึ่งคล้ายกับโค้ดด้านบน

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

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

สมมติว่าคุณมีเฉพาะรายการและรายการดัชนีจริง / ที่จำเป็นซึ่งควรเร็วที่สุด:

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

ซึ่งหมายความว่าการเลือกคุณสมบัติจะทำหลายรอบเท่าที่มีดัชนีจริง / ต้องการเท่านั้น หากคุณมีรายการคุณสมบัติจำนวนมากที่เป็นไปตามกฎของรายการแท็กเดียว (จริง / เท็จ) คุณสามารถสร้างรายการดัชนีโดยใช้หลักการเข้าใจรายการเดียวกัน:

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

สิ่งนี้จะวนซ้ำผ่านแต่ละรายการใน good_objects (ในขณะที่จดจำดัชนีพร้อมแจงนับ) และส่งกลับเฉพาะดัชนีที่รายการนั้นเป็นจริง

สำหรับใครก็ตามที่ไม่เข้าใจรายชื่อนี่คือเวอร์ชันร้อยแก้วภาษาอังกฤษที่เน้นรหัสเป็นตัวหนา:

รายการดัชนีสำหรับกลุ่มทุกดัชนีรายการที่มีอยู่ในแจงนับของวัตถุที่ดี , ถ้า (ที่) เดอะรายการเป็น True

ภาษา Matlab และ Scilab เสนอไวยากรณ์ที่เรียบง่ายและสวยงามกว่า Python สำหรับคำถามที่คุณถามดังนั้นฉันคิดว่าสิ่งที่ดีที่สุดที่คุณทำได้คือเลียนแบบ Matlab / Scilab โดยใช้แพ็คเกจ Numpy ใน Python การทำเช่นนี้วิธีแก้ปัญหาของคุณจะรัดกุมและสวยงามมาก:

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 พยายามเลียนแบบ Matlab / Scilab แต่มีค่าใช้จ่าย: คุณต้องประกาศทุกรายการด้วยคีย์เวิร์ด "array" ซึ่งจะทำให้สคริปต์ของคุณทำงานหนักเกินไป (ปัญหานี้ไม่มีใน Matlab / Scilab) โปรดทราบว่าโซลูชันนี้ จำกัด เฉพาะอาร์เรย์ของจำนวนซึ่งเป็นกรณีในตัวอย่างของคุณ