重塑变量numpy数组的形状

假设我有一个具有给定形状的numpy数组

u

，它是

u

中条目的总数的除数

d

。我如何快速地将

u

重塑为

（某物，d）

u

只是一个双精度的情况也应该包括在内->

（1,1）

---

u

为空的情况应成为

（0，d）

成形数组

您要使用

整形

u.reshape(-1, d)

Python中没有

double

，你是说

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)