python - 使用来自另一个列表的信息生成数字列表

Question

我有一个看起来像这样的文件:

100 2
300 3
50 1
500 5

我想生成一个数字池，其中列表中第一个数字的出现应该出现 x 次(x 是列表中的第二个数字)。

这就是我想要的输出:

[100, 100, 300, 300, 300, 50, 500, 500, 500, 500, 500]

我写了这样一个函数:

def Pool(pos, count):
    pool = pos*int(count)
    return pool

对于每一行，我都将所有数字附加到一个名为 bigpool 的变量中

bigpool = []
for line in weightposfile:
    line = line.rstrip()
    f = line.split('\t')
    pos = f[0]
    count = int(f[1])
    pool = Pool(pos, count)
    bigpool.append(pool)

但是这会返回一个像这样的列表:

[100100, 300300300, 50, 500500500500500]

如何分隔数字并获得我想要的输出(如上所示)？

Answer 1

这应该有效:

def Pool(pos, count):
    return [pos] * count

bigpool = []
for line in weightposfile:
    line = line.rstrip()
    f = line.split('\t')
    pos = f[0]
    count = int(f[1])
    pool = Pool(pos, count)
    bigpool += pool

我改了两行。 return [pos] * count 将生成 pos 的数组。

bigpool += pool 会将 pool 的元素附加到 bigpool。