將一維數組轉換爲nD數組？

Question

我有一個784元素的1維數組，這是一個扁平的28x28圖像。

是否有一個很好的，活潑的，numpy/scipy程序將784元素數組轉換爲28x28數組？

還是我必須做的事：

for i in range(0,28): 
    for j in range(0,28): 
      my_image[i,j] = oneD_vector[28*i + j] 

Answer 1

reshape()作品：

a = np.arange(784) 
a.reshape(28, 28)