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]