Python 列表理解与类成员值

问题描述

我想通过使用两个此类对象列表的成员变量来结束对象列表。所以要选择并结合一个条件。

示例场景，我有类 Frame，带有成员变量 KeyFrameTime：

class Frame:
   def __init__(self):
      self.KeyFrameTime = 0.0

现在有两个包含此类对象的列表

A = Frame()
A.KeyFrameTime = 1.0
B = Frame()
B.KeyFrameTime = 2.0
FooList = [A,B]

C = Frame()
C.KeyFrameTime = 2.0
D = Frame()
D.KeyFrameTime = 3.0
BarList = [C,D]

对于这些列表 FooList 和 BarList，我想使用一些条件并选择满足这些值的对象。例如，如果条件是 KeyFrameTime 相等，我会得到通缉名单：

resultList = []
for f in FooList:
   for b in BarList:
      if(f.KeyFrameTime == b.keyFrameTime):
         resultList.append(f)

这给了我想要的结果列表，但是，我想知道 zip/map/filter 或列表理解是否会以某种方式结束相同的结果。

解决方法

使用 itertools.product() 和列表推导式：

import itertools

resultList = [f for f,b in itertools.product(FooList,BarList) if f.KeyFrameTime == b.KeyFrameTime]