stack

此函数沿新轴连接数组序列。 自NumPy版本1.10.0以来已添加此功能。 需要提供以下参数。

Note - 此功能在version 1.10.0 。

numpy.stack(arrays, axis)

Where,

例子 (Example)

import numpy as np 
a = np.array([[1,2],[3,4]]) 
print 'First Array:' 
print a 
print '\n'
b = np.array([[5,6],[7,8]]) 
print 'Second Array:' 
print b 
print '\n'  
print 'Stack the two arrays along axis 0:' 
print np.stack((a,b),0) 
print '\n'  
print 'Stack the two arrays along axis 1:' 
print np.stack((a,b),1)

它应该产生以下输出 -

First array:
[[1 2]
 [3 4]]
Second array:
[[5 6]
 [7 8]]
Stack the two arrays along axis 0:
[[[1 2]
 [3 4]]
 [[5 6]
 [7 8]]]
Stack the two arrays along axis 1:
[[[1 2]
 [5 6]]
 [[3 4]
 [7 8]]]