Reshape a variable numpy array

Question

Suppose i have a numpy array u with a given shape, a a divisor d of the total number of entries in u. How can i fastly reshape u to be shaped (something,d) ?

The case where u is just a double should be included as well -> (1,1)

The case where u is empty should become a (0,d) shaped array

Nathan Furnal · Accepted Answer · 2020-01-06 13:14:08Z

1

You want to use reshape

u.reshape(-1, d)

There is no double in Python you do you mean float ?

In short :

import numpy as np

def div_reshape(arr, div):
    if arr.size == 0:
        return np.empty(shape=(0, div))
    elif arr.size == 1:
        return arr.reshape(1, 1)
    else:
        return arr.reshape(-1, d)

answered Jan 6, 2020 at 13:14

Nathan Furnal

2,4503 gold badges14 silver badges28 bronze badges

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

Collectives™ on Stack Overflow

Reshape a variable numpy array

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