Python实现取多维数组第n维的前几位-创新互联

现在我们有一个shape为(7352, 9, 128, 1)的numpy数组。

创新互联专业为企业提供朔城网站建设、朔城做网站、朔城网站设计、朔城网站制作等企业网站建设、网页设计与制作、朔城企业网站模板建站服务,10年朔城做网站经验,不只是建网站,更提供有价值的思路和整体网络服务。

想要取出第2维的前三个数据,构成新数组(7352, 3, 128, 1)

我的思想是:将第2维数据转置(transpose)到第一维,再用切片(slice)取出前三个数据,再转置回去:

print("# original", input.shape)
input_transpose = input.transpose((1, 0, 2, 3))
print("# transpose", input_transpose.shape)
input_slice = input_transpose[0:3]
print("# slice", input_slice.shape)
output = input_slice.transpose((1, 0, 2, 3))
print("# output", output.shape)

另外有需要云服务器可以了解下创新互联scvps.cn,海内外云服务器15元起步,三天无理由+7*72小时售后在线,公司持有idc许可证,提供“云服务器、裸金属服务器、高防服务器、香港服务器、美国服务器、虚拟主机、免备案服务器”等云主机租用服务以及企业上云的综合解决方案,具有“安全稳定、简单易用、服务可用性高、性价比高”等特点与优势,专为企业上云打造定制,能够满足用户丰富、多元化的应用场景需求。


新闻标题:Python实现取多维数组第n维的前几位-创新互联
转载来于:http://azwzsj.com/article/deddhc.html