对于一个shape的数组,如何在shape不变的情况下增加一个维度上去,python代码
时间: 2023-06-04 16:08:23 浏览: 96
Python3.5基础之NumPy模块的使用图文与实例详解
可以使用numpy库中的reshape函数来实现,在reshape函数中,将原有的shape作为参数传入,并在其中增加一个维度。示例如下:
import numpy as np
# 原有的数组shape为(2, 3, 4)
a = np.random.rand(2, 3, 4)
# 将原有的shape作为参数,增加一个维度
b = a.reshape((2, 3, 4, 1))
print("原有的shape为:", a.shape)
print("增加一个维度后的shape为:", b.shape)
# 输出结果:
# 原有的shape为: (2, 3, 4)
# 增加一个维度后的shape为: (2, 3, 4, 1)
阅读全文