Python 嵌套 For 循环增强可能吗？

Question

给定一个字符串

abcd

我需要生成以下输出：

['a', 'ab', 'abc', 'abcd', 'b', 'bc', 'bcd', 'c', 'cd', 'd']

我在 Python 3.8 中将以下代码放在一起，它可以工作，但非常笨重且效率低下：

text = 'abcd'
final = [text[y:i+1] for i in range(len(text)) for y in range(len(text)) if len(text[y:i+1]) > 0]
print(sorted(final))

有没有更优雅或更高效的方法来实现所需的输出，而不使用嵌套的 for 循环？

谢谢

Answer 1

试试：

s = "abcd"

out = [s[i:j] for i in range(len(s)) for j in range(i + 1, len(s) + 1)]
print(out)

打印：

["a", "ab", "abc", "abcd", "b", "bc", "bcd", "c", "cd", "d"]

---

编辑：一个循环：

s = "abcd"
n = (len(s) * (len(s) + 1)) // 2

out, cnt1, cnt2 = [], 0, 0
for _ in range(n):
    cnt1 += 1
    out.append(s[cnt2:cnt1])
    if cnt1 > len(s) - 1:
        cnt2 += 1
        cnt1 = cnt2

print(out)

打印：

["a", "ab", "abc", "abcd", "b", "bc", "bcd", "c", "cd", "d"]