numpy column_stack() 用法及代码示例

**numpy.column_stack() **函数用于将一维数组作为列堆叠到二维数组中，它采用一维数组序列并将它们堆叠为列以构成一个二维数组。二维阵列堆叠as-is，就像hstack功能。

**用法：** numpy.column_stack(tup)

参数：
tup :[ndarrays的序列]包含要堆叠的数组的元组。数组必须具有相同的第一维。
Return :[堆叠的二维数组]输入数组的堆叠的二维数组。

代码1：

# Python program explaining 
# column_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  
out_arr = geek.column_stack((in_arr1, in_arr2)) 
print ("Output stacked array:\n ", out_arr)

输出：

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

代码2：

# Python program explaining 
# column_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  
out_arr = geek.column_stack((in_arr1, in_arr2)) 
print ("Output stacked array:\n ", out_arr)

输出：

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

Lincoln普通用户

收藏 海报分享