当前位置: 首页 > 工具软件 > nstack > 使用案例 >

numpy.stack ,nstack,vstack,cat,concatnate,argsort小结

通鸿风
2023-12-01

np.stack

理解:axis参数指的是增加的维度

每次按对应维数取出来相应的数据并加括号组合起来:

如下例:

a = np.array([[1,2,3,4], [5,6,7,8]])
b = np.array([[2,2,1,4], [3,5,7,8]])
c = np.array([[5,7,7,3], [6,6,2,8]])
arrays = np.asarray([a, b , c])
axis=0时,取出a的0维的所有两个元素分别为[1,2,3,4]和[5,6,7,8]组合起来,b,c同样操作,然后把它们组合起来,因为0维之前没有别的维数了,所以只需要一步所有,形状为3*2*4

axis=1时,取出a的第一批一维所有的4个元素为1,2,3,4组合起来得到[1,2,3,4],b,c同样操作,组合起来就是[[1,2,3,4],[2,2,1,4],[5,7,7,3]],然后去a的第二批,最后得到形状2*3*4

axis=2时,取出a的第二维的所有元素1,然后b的2,c的5,组合得到[1,2,5],然后以此类推

 类似资料: