3

我正在寻找一个生成器，它将基于布尔表达式展平一系列元组。我的数据如下所示：

my_data = ((3, 4), (None, 4), (5, 8), (None, 1), (None, 9)...)

我想做的是将它展平为一个一维数字生成器，如果不是，我将在每个元组中取第一项None，否则取第二项。我的结果将产生以下序列：

3, 4, 5, 1, 9...

我认为最简单的方法是使用带有短路功能的布尔表达式，但我似乎无法组成一个合适的生成器。我意识到我可以定义一个生成器函数，这将相当简单，但我很好奇这是否可以通过理解来完成？

我的尝试：

(x or y for subitem in my_data for x, y in subitem)

追溯：

Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "<stdin>", line 1, in <genexpr>
TypeError: 'int' object is not iterable
4

2 回答 2

7 
(x[0] or x[1] for x in my_data)

或者，

(x or y for x,y in my_data)
于 2012-07-23T09:05:36.030 回答
3

恐怕你需要None明确地检查，或者像这样的元组(0, None)或者(0, 1)会让你绊倒：

(x if x is not None else y for x,y in my_data)

例子：

>>> my_data = ((3, 4), (None, 4), (5, 8), (None, 1), (0, None), (0, 1))
>>> [x or y for x,y in my_data]
[3, 4, 5, 1, None, 1]
>>> [x if x is not None else y for x,y in my_data]
[3, 4, 5, 1, 0, 0]
于 2012-07-23T09:15:24.307 回答