我有一个函数foo返回一个形状为(1000,2)的数组
如何将其拆分为两个数组a(1000)和b(1000)
我在寻找这样的东西:
如何将其拆分为两个数组a(1000)和b(1000)
我在寻找这样的东西:
a;b = foo()
我正在寻找一个答案,可以很容易地推广到形状为(1000,5)左右的情况.
解决方法
移调和拆包?
a,b = foo().T
>>> a,b = np.arange(20).reshape(-1,2).T >>> a array([ 0,2,4,6,8,10,12,14,16,18]) >>> b array([ 1,3,5,7,9,11,13,15,17,19])