Question

我有"1-3 6:10-11 7-9"格式的字符串，我想要创建数字集，如下{1,2,3,6,10,11,7,8,9}。

为了从数字范围创建集合，我有以下代码：

def create_set(src):
    lset = []
    if len(src) > 0:
        pos = src.find('-')
        if pos != -1:
            first = int(src[:pos])
            last  = int(src[pos+1:])
        else:
            return [int(src)]  # Only one number
        for j in range (first, last+1): 
            lset.append(j)
        return set(lset)

但是当它出现在字符串中时，我无法弄清楚如何正确对待'：'。有人能帮助我吗？

提前致谢！

编辑：顺便说一句，是否有更简洁的方法来解析这些字符串，可能使用正则表达式？

Answer 1

这样的事可能适合你：

s = '1-3 6:10-11 7-9'
s = s.replace(':', ' ')
lset = set()
fs = s.split()
for f in fs:
    r = f.split('-')
    if len(r)==1:
        # add a single number
        lset.add(int(r[0]))
    else:
        # add a range of numbers (inclusive of the endpoints)
        lset |= set(range(int(r[0]), int(r[1])+1))
print(lset)

Answer 2

编辑：顺便说一句，是否有更简洁的方法来解析这样的字符串，也许使用正则表达式？

也许是一种更清洁（也更有效）的方式：

import re
import itertools

allGroups = re.findall(r"(\d+)(?:-(\d+)|:)", s)
expanded = [range(int(x), (int(x) if y == '' else int(y)) + 1) for x, y in allGroups]
print {x for x in itertools.chain.from_iterable(expanded)}

说明：

匹配所有字符串，如'a-b'或'a：'，并分别返回（a，b）和（a，''）对的列表：

allGroups = re.findall(r"(\d+)(?:-(\d+)|:)", s)

这会产生：

[('1', '3'), ('6', ''), ('10', '11'), ('7', '9')]

使用列表推导将所有（x，y）对扩展到范围（x，y + 1）中的完整数字列表，注意将（x，''）情况处理为（x，x +） 1）：

expanded = [range(int(x), (int(x) if y == '' else int(y)) + 1) for x, y in allGroups]

这会产生：

[[1, 2, 3], [6], [10, 11], [7, 8, 9]]

使用itertools.chain.from_iterable()将列表列表转换为单个iterable，并通过set comprehension迭代到最终集合中：

print {x for x in itertools.chain.from_iterable(expanded)}

这会产生：

set([1, 2, 3, 6, 7, 8, 9, 10, 11])

Python - 从字符串

2 个答案: