在 Python 中将序列拆分为单个步骤的最优雅方法答案

【问题标题】：Most elegant way to split a sequence into individual steps in Python在 Python 中将序列拆分为单个步骤的最优雅方法
【发布时间】：2021-11-16 04:07:46
【问题描述】：

假设我有一个从 0 到 9 的列表：

lst = list(range(10))

我想将其拆分为 2 个单独的步骤。我设法编写了以下工作代码：

res = [[] for _ in range(len(lst) - 1)]
for i, x in enumerate(lst):
    if i < len(lst) - 1:
        res[i].append(x)
    if i > 0:
        res[i-1].append(x)

>>> print(res)
[[0, 1], [1, 2], [2, 3], [3, 4], [4, 5], [5, 6], [6, 7], [7, 8], [8, 9]]

但我觉得应该有更优雅的编码方式。有什么建议吗？

【问题讨论】：

标签： python python-3.x list split chunks

【解决方案1】：

您可以将 zip 函数与列表理解一起使用。

>>> lst = list(range(10))
>>> [list(elm) for elm in zip(lst, lst[1:])]
[[0, 1], [1, 2], [2, 3], [3, 4], [4, 5], [5, 6], [6, 7], [7, 8], [8, 9]]

【讨论】：

【解决方案2】：

对于元组列表：

>>> list(zip(lst,lst[1:]))
[(0, 1), (1, 2), (2, 3), (3, 4), (4, 5), (5, 6), (6, 7), (7, 8), (8, 9)]

对于列表列表：

>>> list(map(list, zip(lst,lst[1:])))
[[0, 1], [1, 2], [2, 3], [3, 4], [4, 5], [5, 6], [6, 7], [7, 8], [8, 9]]

【讨论】：

感谢@AlainT。已编辑。
@AlainT。我的评论是针对使用zip(lst[:-1],lst[1:]) 的先前版本。现在这很常见而且很好。

【解决方案3】：

可以概括为n步骤：

def steps(lst, n=2):
    return [[*x] for x in zip(*(lst[i:] for i in range(n)))]

steps(range(10), 2)
# [[0, 1], [1, 2], [2, 3], [3, 4], [4, 5], [5, 6], [6, 7], [7, 8], [8, 9]]
steps(range(10), 3)
# [[0, 1, 2], [1, 2, 3], [2, 3, 4], [3, 4, 5], [4, 5, 6], [5, 6, 7], [6, 7, 8], [7, 8, 9]]

【讨论】：

这是一个优雅的通用解决方案！

【解决方案4】：

这是一个没有zip的解决方案，只使用索引，range、len和list comprehension：

lst = list(range(10))
res = [ [lst[i], lst[i+1]] for i in range(len(lst) - 1)]
print(res)
# [[0, 1], [1, 2], [2, 3], [3, 4], [4, 5], [5, 6], [6, 7], [7, 8], [8, 9]]

【讨论】：

【解决方案5】：

您可以完全避免zip，只需使用列表推导来创建滑动窗口。

[lst[i: i + 2] for i in range(0, len(lst) - 1)]
>> [[0, 1], [1, 2], [2, 3], [3, 4], [4, 5], [5, 6], [6, 7], [7, 8], [8, 9]]

【讨论】：