numpy中包含的newaxis可以给原数组增加一个维度,np.newaxis放置的位置不同,产生的新数组也不同
import numpy as np
x = np.random.randint(1, 8, size=(2, 3, 4))
y = x[np.newaxis, :, :, :]
z = x[:, np.newaxis, :, :]
h = x[:, :, np.newaxis, :]
m = x[:, :, :, np.newaxis]
总结:想放哪里放哪里,想放几个放几个