Reshaping python array for unknown rows using -1 & filling remaining spaces with None

Question

I have an array with unknown length (for example lets use 11). so the array is

[1,2,3,4,5,6,7,8,9,10,11]

I want to reshape that array so he will have 5 columns and as much rows as needed. I know I can use reshape(-1,5) that way it creates the rows based on the array length.

But it gives me this error:

ValueError: cannot reshape array of size 11 into shape (5)

Any idea how do I do that? the desired result is:

[[1,2,3,4,5],
[6,7,8,9,10],
[11,None,None,None,None]]

The code that I ran and got this error is:

import numpy as np
a = np.array([1,2,3,4,5,6,7,8,9,10,11])    
print(np.reshape(a, (-1,5)))

Dup of stackoverflow.com/questions/41253044/… ?

crayxt
– crayxt

2019-08-25 10:42:49 +00:00
Commented Aug 25, 2019 at 10:42 — crayxt
– crayxt, Commented Aug 25, 2019 at 10:42

Olvin Roght · Accepted Answer · 2019-08-25 10:44:43Z

1

You can do it without numpy.

ar = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11]
n = 5
reshaped = [ar[i: i + n] + [None] * (i + n - len(ar)) for i in range(0, len(ar), n)]

You can also use trick with iterator (chunks will be in tuples):

reshaped = list(zip(*[iter(ar + [None] * (n - len(ar) % n))] * n))

You can apply zip_longest() from itertools to not add None values by yourself:

from itertools import zip_longest
reshaped = list(zip_longest(*[iter(ar)] * n))

edited Aug 25, 2019 at 10:44

answered Aug 25, 2019 at 10:16

Olvin Roght

7,8432 gold badges19 silver badges39 bronze badges

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

5 Comments

orr Over a year ago

Is there a more efficient way? maybe one numpy function or withput loops?

Olvin Roght Over a year ago

@orr. what do you mean under "efficient"? You can't split list to chunks without loops. I'm not specialist in numpy, so I don't know it's methods.

orr Over a year ago

by efficient I mean built-in numpy functions so that will make faster runtime. your answer is great, but I am curious to knoe whether there is faster way.

Olvin Roght Over a year ago

@orr, if tests from this answer is correct, numpy 3 times slower then list comprehension.

orr Over a year ago

haha thats funny, I thought numpy should be more faster. I guess I was wrong, thanks for the enlightenment.

hpaulj · Accepted Answer · 2019-08-25 15:06:37Z

0

In [135]: res = np.empty((3,5), object)                                                                      
In [136]: res                                                                                                
Out[136]: 
array([[None, None, None, None, None],
       [None, None, None, None, None],
       [None, None, None, None, None]], dtype=object)
In [137]: res.flat[:11] = np.arange(1,12)                                                                    
In [138]: res                                                                                                
Out[138]: 
array([[1, 2, 3, 4, 5],
       [6, 7, 8, 9, 10],
       [11, None, None, None, None]], dtype=object)

answered Aug 25, 2019 at 15:06

hpaulj

233k14 gold badges260 silver badges392 bronze badges

2 Comments

orr Over a year ago

the thing is I dont know the row number, so I dont know the size of the empty None array I need. I only know the columns

hpaulj Over a year ago

But you can calculate the row number. The next multiple of 5 up from 11 is 15. 15/5 gives me 3.

Collectives™ on Stack Overflow

Reshaping python array for unknown rows using -1 & filling remaining spaces with None

2 Answers 2

5 Comments

2 Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

5 Comments

2 Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related