用Python从数组中打印N次

Question

我见过几个多次打印字符串的示例，但找不到多次打印 int 的解决方案。

我有两个数组。

我想打印 1, 1 次 3,2 次和 4, 2 次。 我知道我不能像打印字符串 n 次那样使用 * 运算符

我怎么去

[1,3,3,4,4]

我写的代码如下：

A = [1, 3, 4]
B = [1, 2, 2]
i = 0
j = 0
for i in range(0, len(A), 1):
    for j in range(0, len(B), 1):
        if i==j:
            print A[i] * B[j]

更新：我编辑了代码以反映实际数字，因为我测试了一个代码并发布了另一个代码。对此感到抱歉。

Answer 1

列表推导的另一种选择

[e for e,c in zip(a,b) for i in range(c)]
[1, 3, 3, 4, 4]

或者，更hacky

sum([[e]*c for e,c in zip(a,b)],[])
[1, 3, 3, 4, 4]

Answer 2

您可以使用列表推导扩展一个空列表。将您的两个列表 A 和 B 压缩在一起，以并行处理它们。

result = []
_ = [result.extend([a] * b) for a, b in zip(A, B)]

>>> result
[1, 3, 3, 4, 4]

编辑

列表推导应该只用于创建列表，而不是它的副作用。最好在上面的示例中使用循环。

result = []
for a, b in zip(A, B):
    result.extend([a] * b)

Answer 3

您的代码中有很多您不需要的额外内容。

A=[1,3,4]
B=[1,2,2]
"""first loop over the elements you want to print in A"""
for i in range(len(A)):
    """ A[i] j times """
    for j in range(B[i]):
        print A[i]

我建议阅读一下范围函数在 python 中的工作原理https://docs.python.org/2/library/functions.html#range

如果您想将这些值添加到数组中，您必须先定义该数组，然后将它们添加到其中而不是打印它们。

A=[1,3,4]
B=[1,2,2]
result=[]
"""first loop over the elements you want to print in A"""
for i in range(len(A)):
    """ append A[i] to result j times """
    for j in range(B[i]):
        result.append(A[i])

print str(result)