Question

我想在Swift中为String类创建一个扩展，它允许你像Python一样通过下标运算符获取子字符串。这可以通过以下方式使用Range类来完成

extension String {

    subscript (range: Range<Int>) -> String? {
        if range.startIndex < 0 || range.endIndex > count(self) {
            return nil
        }
        let range = Range(start: advance(startIndex, range.startIndex), end: advance(startIndex, range.endIndex))
        return substringWithRange(range)
    }
}

这可以实现以下用途

let string = "OptimusPrime"
string[0...6] // "Optimus"

但是，我希望能够使用负整数从结尾和开头索引字符串。

string[7...(-1)] // "Prim"

由于Range类不允许startIndex大于endIndex，因此上述扩展不足以实现此行为。

由于Range对象显然无法用于此扩展，因此我只需使用与Python相同的语法就可以了

string[0:2] // "Op"
string[0:-1] // "OptimusPrim"

这可能吗？我该怎么做呢？

Answer 1

我认为你可以像这样获得类似python的语法：

subscript (start:Int, end:Int) -> String?

可以让你去"bla bla"[0,3]

然后如果您输入负数，只需将其转换为字符串长度减去负数，以便范围有效

复杂的Swift扩展

1 个答案: