时间戳

首页 > 解决方案 > 如何在 Python 3 中使用生成器管道？

python - 如何在 Python 3 中使用生成器管道？

问题描述

我正在努力提高我的能力并学习生成器管道。我设法解决了一些问题，但我觉得我的理解存在基本差距。

例如：我试图构建一个返回范围内所有数字的函数。以几个范围的字符串形式给出的范围。所以收到的字符串的一个例子："1-2,4-4,8-10". 所以第一个生成器应该返回一个 int 对的列表：[1, 2] [4, 4] [8, 10] 第二个生成器应该在一个范围函数中使用1st_gen[0]as start 和 the1st_gen[1]作为 stop 并返回范围内的所有数字：1 2 4 8 9 10

有我的代码，我很乐意提供提高我技能的提示：

def parse_ranges(range_string):
    temp_list = (c.replace("-", ",") for c in list(range_string.split(",")))
    generator2 = (i for start, stop in temp_list for i in range(int(start), int(stop) + 1))
    for i in generator2:
        print(i)

print(parse_ranges("1-2,4-4,8-10"))
print(parse_ranges("0-0,4-8,20-21,43-45"))

标签： pythonpython-3.xgenerator

解决方案

一些修复：

  • str.split()返回一个列表，无需将其结果放入list
  • parse_ranges()不返回任何东西，为什么要打印它？
  • range(int(4), int(4) + 1)不会回来[4, 4]，但只会[4]
def parse_ranges(range_string):
    temp = (map(int, c.split("-")) for c in range_string.split(",")) 

    # yield all the numbers (yield from 'flattens' the result)
    for start, stop in temp:
        yield from range(start, stop + 1)

    # or to yield lists 
    # for start, stop in temp:
    #     yield list(range(start, stop + 1))

print(*parse_ranges("1-2,4-4,8-10"), sep=" ~ ")
print(*parse_ranges("0-0,4-8,20-21,43-45"), sep=" ~ ")

输出（'~'作为分隔符添加，因此很清楚，产生的元素的“边界”在哪里）：

# yield from range(start, stop + 1)
1 ~ 2 ~ 4 ~ 8 ~ 9 ~ 10
0 ~ 4 ~ 5 ~ 6 ~ 7 ~ 8 ~ 20 ~ 21 ~ 43 ~ 44 ~ 45

# yield list(range(start, stop + 1))
[1, 2] ~ [4] ~ [8, 9, 10]
[0] ~ [4, 5, 6, 7, 8] ~ [20, 21] ~ [43, 44, 45]

推荐阅读