重塑可变 numpy 数组

Question

假设我有一个给定形状的 numpy 数组 u，u 中条目总数的除数 d。我怎样才能快速重塑 u 以塑造 (something,d) ？

u 只是双精度数的情况也应该包括在内 -> (1,1)

u为空的情况应该变成(0,d)形数组

Answer 1

您想使用reshape

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)

重塑可变 numpy 数组

Reshape a variable numpy array

numpy

reshape