Turn a 1D array into nD array?

Question

I have a 784 element 1-dimensional array that is a flattened 28x28 image.

Is there a nice, snappy, numpy/scipy procedure for translating that 784 element array into a 28x28 array?

Or do I have to do:

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

have you seen numpy.reshape: http://docs.scipy.org/doc/numpy-1.10.1/reference/generated/numpy.reshape.html — Garrett R, Feb 05 '16 at 23:22

Mike Müller · Answer 1 · 2016-02-05T23:26:47.977

4

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

edited Feb 05 '16 at 23:26

answered Feb 05 '16 at 23:23

Mike Müller

1 Answers1