Python:使用包含索引值的列表来拆分另一个列表



我有一个元组列表,表示形状的坐标-"点"

我还有一个索引值列表,用于确定该形状每个部分的起点(因为形状可以包含一个或多个"岛")-"部分"

目前,为了从点列表中提取零件的点,我执行以下操作:

points = [
    (0, 0), (1, 0), (2, 1), (3, 2), (0, 0),
    (4, 5), (5, 4), (3, 6), (2, 9), (4, 5),
    (2, 2), (3, 2), (6, 4), (4, 4), (8, 6), (4, 3), (2, 2)
]
parts = [0, 5, 10]
for index, part in enumerate(parts):
    start = part
    try:
        end = parts[index + 1]
    except Exception:
        end = len(points)
    print(points[start:end])

然而,这对我来说并不是特别的Python。老实说,使用异常只是感觉非常肮脏。有人能为这个特殊的问题提出一个改进的Pythonv3方法吗?

您想要这样做:

parts = [0, 5, 10, len(points)]
start_end_pairs = zip(parts[:-1], parts[1:])
for start, end in start_end_pairs:
    print(points[start:end])

解释

您的parts已经包含起始索引,因此您只需要获取该数组的内容即可计算所需的索引。

[0, 5, 10]开始,第一部分需要取(0, 5),第二部分取(5, 10),第三部分取(10, len(points))。创建这些对以在print-循环中使用的过程可以很容易地用zip来实现。我压缩了两个阵列:

[0, 5, 10]
[5, 10, len(points)]

生产

[(0, 5), (5, 10), (10, len(points))]

这正是你所需要的。

for i in range(len(parts)):
    if i==len(parts)-1:
        print(points[parts[i]:])
    else:
        print(points[parts[i]:parts[i+1]])

或:

for i in range(len(parts)-1):
    print(points[parts[i]:parts[i+1]])
print(points[parts[-1]:])

或:

for start, end in zip(parts[:-1], parts[1:]):
    print(points[start:end])
print(points[parts[-1]:])

最新更新