我需要从 Python 中的元组打印/返回多个元素

Question

我有一个字符串和元组的列表。元组看起来像这样（数字、元素）：

s = ["element1", (3, "element2"), (2, "element3")]

我需要像这样打印元组：

element1
element2
element2
element2
element3
element3

也就是说，我需要数字*元素打印。我想出的最好的是：

s = ["element1", (3, "element2"), (2, "element3")]
def bb (x):
    for n in x:
        if isinstance (n, str):
           print (n)
        if isinstance (n, tuple):
            print (n[1])

Answer 1

理想情况下，更改您的输入以便获得[(1, "element1,"), ...]，但如果不可能：

import types
import itertools
s = ["element1", (3, "element2"), (2, "element3")]
print(list(itertools.chain.from_iterable((e, ) if isinstance(e, str) else itertools.repeat(e[1], e[0]) for e in s)))

为我们提供：

['element1', 'element2', 'element2', 'element2', 'element3', 'element3']

我们在这里所做的是获取您的元素并将它们转换为元组 if 字符串（这样它们就不会迭代为字符），并使用 itertools.repeat() 创建一个迭代器，如果它们是元组，则重复给定的次数。然后，我们使用 itertools.chain.from_iterable() 将此集合平展为单个迭代器，然后将其转换为列表。

请注意，在 2.x 中，您希望isinstance(e, basestring) .

编辑：或者：

args = ((e, 1) if isinstance(e, str) else (e[1], e[0]) for e in s)
itertools.chain.from_iterable(itertools.starmap(itertools.repeat, args))

这是一回事，但在我看来可能更整洁。

Answer 2

这是一个单行：

s = ["element1", (3, "element2"), (2, "element3")]
result = sum(([x] if type(x) != tuple else x[0]*[x[1]] for x in s), [])

或者扩展现有代码：

def bb(x):
    for n in x:
        if isinstance(n, str):
            print(n)
        elif isinstance(n, tuple):
            count, value = n
            for i in range(count):
                print(value)

如果要返回这些值，可能需要使用生成器：

def bb_gen(x):
    for n in x:
        if isinstance(n, str):
            yield n
        elif isinstance(n, tuple):
            count, value = n
            for i in range(count):
                yield value
for value in bb_gen(s):
    print value