Python:根据索引集从列表中选择子集

Question

我有几个列表具有相同数量的条目(每个列表指定一个对象属性):

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]

Answer 1

你可以使用列表理解:

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(good_objects, property_a))

Answer 2

我看到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]
   

Answer 3

使用内置功能zip

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

Answer 4

假设您只有项目列表和真实/必需索引列表,这应该是最快的:

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

这意味着属性选择将只执行与真实/必需索引一样多的轮次.如果您有许多属性列表遵循单个标记(true/false)列表的规则,则可以使用相同的列表推导原则创建索引列表:

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

这将遍历good_objects中的每个项目(同时使用枚举记住其索引)并仅返回item为true的索引.

对于没有获得列表理解的任何人,这里是一个英文散文版本,代码以粗体突出显示:

列出每个索引组的索引,该项目存在于良好对象的枚举中,if(where)该项为True