flatten array of arrays into a vector of arrays

Question

I'm generating matrix representations of images with height*width size, and I need to transform them in a vector of pixels. To generate the images, I'm using the following instruction

np.array([[np.random.randint(0, 255, 3) for dummy_row in range(height)] for dummy_col in range(width)])

e.g., (2x2) image

array([[[132, 235,  40],
        [234,   1, 160]],
       [[ 69, 108, 218],
        [198, 179, 165]]])

I tried to use flatten(), but is not creating a one-dimension array of pixels, but is pilling all the values together

array([132, 235,  40, 234,   1, 160,  69, 108, 218, 198, 179, 165])

when I'm requiring

array([132, 235,  40], [234,   1, 160],  [69, 108, 218], [198, 179, 165]])

is there a built-in function to get this output?

norok2 · Accepted Answer · 2020-01-27 21:27:59Z

1

Just use:

arr.reshape(-1, n_channels)

or similar (where arr is the NumPy array containing the image data).

answered Jan 27, 2020 at 21:27

norok2

27.1k6 gold badges83 silver badges110 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

flatten array of arrays into a vector of arrays

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related