python - 如何在一个列表中乘以元素，同时在另一个列表中提供范围

Question

我有两个列表，稍后我将使用它们来确定我正在查看文档的页数。第一个列表 (l_name) 包含文档的名称。第二个列表 (l_depth) 包含我要查看的页数，总是从第一个开始。

原始列表如下所示:

l_name = ['Doc_1', 'Doc_2', 'Doc_3']
l_depth = [1, 3, 2]

因为我想使用 for 循环来指示我将要查看的每个页面

for d,p in zip(l_doc, l_page):
  open doc(d) on page(p)
  do stuff

我需要新列表看起来像这样:

l_doc = ['Doc_1', 'Doc_2', 'Doc_2', 'Doc_2', 'Doc_3', 'Doc_3']
l_page = [1, 1, 2, 3, 1, 2]

如何根据所需的深度乘以名称 (l_name --> l_doc) 并根据深度提供范围 (l_depth --> l_page)？

Answer 1

你可以通过理解得到你的列表:

[k for i, j in zip(l_name, l_depth) for k in [i]*j]