今天小编就为大家分享一篇Python 实现取多维数组第n维的前几位，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧

现在我们有一个shape为(7352, 9, 128, 1)的numpy数组。 想要取出第2维的前三个数据，构成新数组(7352, 3, 128, 1) 我的思想是：将第2维数据转置（transpose）到第一维，再用切片（slice）取出前三个数据，再转置回去： print(# original, input.shape) input_transpose = input.transpose((1, 0, 2, 3)) print(# transpose, input_transpose.sh