2 回答

TA貢獻1804條經驗 獲得超2個贊
一些修復:
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]

TA貢獻2019條經驗 獲得超9個贊
如果需要,您可以將流程拆分為任意多個生成器步驟,例如:
def get_ranges(string):
for part in string.split(","):
yield [int(number) for number in part.split("-")]
def get_numbers(ranges):
for start, stop in ranges:
yield from range(start, stop + 1)
用法:
>>> list(get_numbers(get_ranges("1-2,4-4,8-10")))
[1, 2, 4, 8, 9, 10]
如果您確實愿意,您甚至可以將其拆分為更多步驟(拆分為多個部分,轉換為數字,轉換為范圍對象,獲取數字)。
添加回答
舉報