python - LineCollection 的 Numpy zip 数组

Question

是否有一种简单的方法来压缩 numpy 列以用作 LineCollection 的输入？

当前格式

请注意，数组的元素比将要使用的元素多得多。

arr=([(x1, y1, z1, a1, b1, c1), (x2, y2, z2, a2, b2, c2), (x3, y3, z3, a3, b3, c3)])

所需格式

我想获取以下格式的行，以便我可以使用 LineCollection:

lines = [[(x1, x2), (y1, y2)], [(x2, x3), (y2, y3)] ...]
LineCollection(lines)

Answer 1

我不知道任何直接的 numpy 函数或类似的函数可以解决这个问题。

但是如果您想要的只是每个子列表的前两个元素，那么您可以尝试 -

arr = [(1,2,3,),(2,3,4),(3,4,5),(4,5,6),(5,6,7),(6,7,8)]
lines = []
for i in range(len(arr)-1):
    lines.append([(arr[i][0],arr[i+1][0]),(arr[i][1],arr[i+1][1])])

lines
>>> [[(1, 2), (2, 3)], [(2, 3), (3, 4)], [(3, 4), (4, 5)], [(4, 5), (5, 6)], [(5, 6), (6, 7)]]