python - 从堆叠数组中获取所有 x 值？ Python

Question

我得到了一个包含 x,y 坐标的 2x1 数组列表。所以它看起来像[np.array([x,y]), np.array([x,y]), np.array([x,y]), ...]

现在我想要一个列表或数组，其中包含所有 x 值和一个包含所有 y 值的列表或数组。但是像 list[:][1] 这样的事情是行不通的。

知道如何做到这一点吗？

Answer 1

>>> import numpy as np
>>> L = [np.array([1, 2]), np.array([3, 4]), np.array([5, 6])]
>>> X = [a[0] for a in L]
>>> Y = [a[1] for a in L]
>>> X
[1, 3, 5]
>>> Y
[2, 4, 6]

正如 @DSM 指出的，您也可以转置列表:

>>> X, Y = map(list, zip(*L))
>>> X
[1, 3, 5]
>>> Y
[2, 4, 6]