C/C++教程

np.stack() 堆叠一维数组时用法

本文主要是介绍np.stack() 堆叠一维数组时用法,对大家解决编程问题具有一定的参考价值,需要的程序猿们随着小编来一起学习吧!

import numpy as np

a = np.array([i for i in range(5)])
b = np.array([i for i in range(5, 10)])
print('a:', a)
print('b:', b)
c = np.stack([a,b], axis=0)
print(c)
d = np.stack([a,b], axis=1)
print(d)
e = np.stack([a,b], axis=-1)
print(e)

结果:
a:   [0 1 2 3 4]

b:   [5 6 7 8 9]

c:   [[0 1 2 3 4]     # axis=0
       [5 6 7 8 9]]

d:   [[0 5]     # axis=1
       [1 6]
       [3 8]
       [4 9]]

e:   [[0 5]     # axis=-1
       [1 6]
       [3 8]
       [4 9]]

这篇关于np.stack() 堆叠一维数组时用法的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!