从堆叠阵列中获取所有x值？蟒蛇

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]