numpy.stack


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

- 此功能在 1.10.0版本中可用

numpy.stack(arrays, axis)

这里

序号 参数和描述
1 arrays 相同形状的阵列序列
2 axis 结果阵列中的轴,输入阵列堆叠在其中

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]]]