如何调整一个浮点数组的大小？

Question

我想调整我的数组(包含浮点数)x_initial的大小，它的形状是(2382, 256, 256)到(2382, 256, 256,1)。

当我使用

x_initial = x_initial.resize(x_initial[0], x_initial[1], x_initial[2], 1)

我得到一个错误

TypeError: only integer scalar arrays can be converted to a scalar index.

有什么方法可以调整一个浮动数组的大小？

Answer 1

这是因为 reshape() 期望输入整数索引，而你输入的是浮点数。试试这样：-

x_initial = x_initial.resize(int(x_initial[0]), int(x_initial[1]), int(x_initial[2]), 1)