python - 安全转换为元组/列表

Question

我有一个元组 y = (3,4)，我想将其附加到列表 z 的每个元素。

如果 z 包含单个元素我可以这样做:

>>> z = [1, 2]
>>> [(x, ) + y for x in z]
[(1, 3, 4), (2, 3, 4)]

如果 z 包含元组，我可以这样做:

>>> z = [(0, 1), (1, 2)]
>>> [x + y for x in z]
[(0, 1, 3, 4), (1, 2, 3, 4)]

但是如果我事先不知道z 是单个元素还是元组/列表，我该如何继续？

我试过:

>>> z = [1, 2]
>>> [tuple(x) + y for x in z]

但这给了我:

TypeError: 'int' object is not iterable

显然我可以创建一个函数来测试元组:

In [38]: def tuplify(x):
   ....:     try:
   ....:         return tuple(x) # This is 'transparent' if x is already a tuple
   ....:     except TypeError:
   ....:         return (x, )

但这似乎很不雅观。什么是继续的好方法？

注意:整个问题同样可以针对列表提出

Answer 1

如果你真的想要一个单行，这个:

[x + y if isinstance(x, tuple) else (x, ) + y for x in z]

将为您工作。

这是同时尝试列表和元组列表的结果:

>>> y = (3,4)
>>> z = [1, 2]
>>> [x + y if isinstance(x, tuple) else (x, ) + y for x in z]
[(1, 3, 4), (2, 3, 4)]
>>> z = [(0, 1), (1, 2)]
>>> [x + y if isinstance(x, tuple) else (x, ) + y for x in z]
[(0, 1, 3, 4), (1, 2, 3, 4)]

如果 y 是一个列表，这将起作用:

[x + y if isinstance(x, list) else [x] + y for x in z]