Изменить форму массива numpy

Question

Предположим, у меня есть numpy массив u с заданной формой, делитель d от общего числа записей в u. Как я могу быстро изменить u, чтобы иметь форму (something,d)?

Также следует включить случай, когда u - просто двойное число -> (1,1)

Случай, когда u пусто должно стать (0,d) массивом

Answer 1

Вы хотите использовать reshape

u.reshape(-1, d)

Нет double в Python Вы имеете в виду float?

Короче говоря:

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)