Python 将 Counter 对象转换为序数列表

Question

我认为这应该有一个相对简单的解决方案。

我有一个 Counter 对象，我想从中创建一个由整数组成的列表，这些整数表示 Counter 对象中每个键的序数值。

我的意思的说明：

假设这是我的反对象 -

Counter(elems)

Counter({'elem1': 13,
         'elem2': 13})

由此，我想生成以下类型的“序数”列表：

[1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13]

我尝试了一些方法，例如：

ord_list = [i for i in range(1, v + 1) for k, v in Counter(elems).items()]

但我一直得到的最接近的是

[1, 1, 2, 2, 3, 3, 4, 4..., 13, 13]

，而不是我正在寻找的。

我错过了什么明显的东西？谢谢大家。

Answer 1

您缺少第二个循环，该循环将每个范围中的每个单独的数字添加到外部列表中。另外，请记住，不需要迭代

items

，因为您不在任何地方使用这些键，因此只需迭代

values

。

d = {'elem1': 13,
     'elem2': 13}

print([num for v in d.values() for num in range(1, v + 1)])

或者您可以使用

itertools.chain

来展平子列表，在我看来，这是一个更长的解决方案，但更具可读性/描述性：

from itertools import chain

d = {'elem1': 13,
     'elem2': 13}

print(list(chain.from_iterable(range(1, v + 1) for v in d.values()))) 

Answer 2

这是您需要的：

elems = {'elem1': 13, 'elem2': 13}
ord_list = [i for k, v in Counter(elems).items() for i in range(1, v + 1)]
print(ord_list)

输出：

[1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13]