Python numpy.stack()用法及代碼示例

numpy.stack()函數用於沿新軸連接相同尺寸數組的序列。axis參數指定結果軸尺寸中新軸的索引。例如，如果axis = 0，它將是第一個尺寸；如果axis = -1，它將是最後的尺寸。

用法： numpy.stack(arrays, axis)

參數：
arrays :[數組]相同形狀的數組序列。
axis :[int]輸入數組沿其堆疊的結果數組中的軸。

Return :[stacked ndarray]輸入數組的堆棧數組，其維數比輸入數組大。

代碼1：

# Python program explaining 
# stack() function 
  
import numpy as geek 
  
# input array 
in_arr1 = geek.array([ 1, 2, 3] ) 
print ("1st Input array:\n", in_arr1)  
  
in_arr2 = geek.array([ 4, 5, 6] ) 
print ("2nd Input array:\n", in_arr2)  
  
# Stacking the two arrays along axis 0 
out_arr1 = geek.stack((in_arr1, in_arr2), axis = 0) 
print ("Output stacked array along axis 0:\n ", out_arr1) 
  
# Stacking the two arrays along axis 1 
out_arr2 = geek.stack((in_arr1, in_arr2), axis = 1) 
print ("Output stacked array along axis 1:\n ", out_arr2)

輸出：

1st Input array:
 [1 2 3]
2nd Input array:
 [4 5 6]
Output stacked array along axis 0:
  [[1 2 3]
 [4 5 6]]
Output stacked array along axis 1:
  [[1 4]
 [2 5]
 [3 6]]

代碼2：

# Python program explaining 
# stack() function 
  
import numpy as geek 
  
# input array 
in_arr1 = geek.array([[ 1, 2, 3], [ -1, -2, -3]] ) 
print ("1st Input array:\n", in_arr1)  
  
in_arr2 = geek.array([[ 4, 5, 6], [ -4, -5, -6]] ) 
print ("2nd Input array:\n", in_arr2)  
  
# Stacking the two arrays along axis 0 
out_arr1 = geek.stack((in_arr1, in_arr2), axis = 0) 
print ("Output stacked array along axis 0:\n ", out_arr1) 
  
# Stacking the two arrays along axis 1 
out_arr2 = geek.stack((in_arr1, in_arr2), axis = 1) 
print ("Output stacked array along axis 1:\n ", out_arr2) 
  
# Stacking the two arrays along last axis  
out_arr3 = geek.stack((in_arr1, in_arr2), axis = -1) 
print ("Output stacked array along last axis:\n ", out_arr3)

輸出：

1st Input array:
 [[ 1  2  3]
 [-1 -2 -3]]
2nd Input array:
 [[ 4  5  6]
 [-4 -5 -6]]
Output stacked array along axis 0:
  [[[ 1  2  3]
  [-1 -2 -3]]

 [[ 4  5  6]
  [-4 -5 -6]]]
Output stacked array along axis 1:
  [[[ 1  2  3]
  [ 4  5  6]]

 [[-1 -2 -3]
  [-4 -5 -6]]]
Output stacked array along last axis:
  [[[ 1  4]
  [ 2  5]
  [ 3  6]]

 [[-1 -4]
  [-2 -5]
  [-3 -6]]]

相關用法

注：本文由純淨天空篩選整理自jana_sayantan大神的英文原創作品 numpy.stack() in Python。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。