主页 / 问题 / 817236
Accepted
pynista
Asked:2020-04-21 18:40:56 +0000 UTC2020-04-21 18:40:56 +0000 UTC

生成器：从嵌套列表中输出第一对，然后是最后一个

  • 772

我有一个包含数字列表的列表。每个内部列表的长度为 4。例如：

list_with_ids = [[123, 234, 345, 456], [678, 890, 901, 1001], [2000, 3000, 4000, 5000],]

如何使生成器从内部列表中提供数字，如下所示：

123->234->678->890->2000->3000->345->456->901->1001->4000->5000

那些。首先，依次给出索引为 0 和 1 的内部列表中的数字，然后，当它们用完时，索引为 2 和 3 的数字也同样给出？

python

2 个回答

  1. Best Answer

    如果您决定在额头上，那么您可以通过某种方式：

    def myGenerator(list):
    for el in list:
        yield el[0]
        yield el[1]
    for el in list:
        yield el[2]
        yield el[3]
    • 2 
  2. from itertools import chain 
from operator import itemgetter

g = chain(map(itemgetter(0, 1), lst),
          map(itemgetter(-2, -1), lst))
    • 1

相关问题