将数组大小调整为 Python 中的另一个形状

Question

我有以下数组：

a = np.random.rand(5,2)
a

array([[0.98736372, 0.07639041],
       [0.45342928, 0.4932295 ],
       [0.75789786, 0.48546238],
       [0.85854235, 0.74868237],
       [0.13534155, 0.79317482]])

并且我想调整它的大小，以便将其分为 2 个批次，包含三个元素（根据需要添加零）：

array([[[0.98736372, 0.07639041],
       [0.45342928, 0.4932295 ],
       [0.75789786, 0.48546238]],
       [[0.85854235, 0.74868237],
       [0.13534155, 0.79317482],
       [0, 0]]])

我已经试过了，但是 returns None:

a = a.copy()
a.resize((2,3,2), refcheck = False)

我相信 .reshape 不会提供解决方案，因为它无法用 0 填充以符合数组所需的尺寸。

Answer 1

使用numpy.resize，你必须这样使用：

import numpy as np
a = np.random.rand(5,2)
b = np.resize(a, (2,3,2))

否则你可以使用对象方法得到相同的结果，像这样：

import numpy as np
a = np.random.rand(5,2)
a.np.resize(2,3,2)
b = a.copy()

注意第一个 return ndarray 和最后一个 returns None 因为它改变了对象本身。有关详细信息，请查看 numpy.resize doc

将数组大小调整为 Python 中的另一个形状

Resizing an array to another shape in Python

python

arrays

resize

numpy

reshape