使用索引重建numpy数组

Question

我有一个numpy个元素的3*k数组，其中k是一个整数。例如，对于k=3，我有一个数组A。下面的x，y和z由A中的绘图元素填充，如下例所示：

import numpy as np

A = np.arange(9)
>> array([0, 1, 2, 3, 4, 5, 6, 7, 8])

x = A[0::3]
>> array([0, 3, 6])

y = A[1::3]
array([1, 4, 7])

z = A[2::3]
>> array([2, 5, 8])

现在，我想使用x，y和z来重建A。有没有使用for循环的简便方法？

Answer 1

您可以

np.array([x,y,z]).T.reshape(-1)