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

當我們想要將輸入轉換為至少二維數組時，使用numpy.atleast_2d()函數。標量和一維輸入被轉換為二維數組，而高維輸入被保留。

用法： numpy.atleast_2d(*arrays)

參數：
arrays1，arrays2，...：[數組]一個或多個array-like序列。非數組輸入將轉換為數組。保留已經具有二維或更多維的數組。

Return :一個數組或數組列表，每個數組的a.ndim> =2。在可能的情況下，避免複製，並返回具有二維或更多維的視圖。

代碼1：工作

# Python program explaining 
# numpy.atleast_2d() function 
  
import numpy as geek 
in_num = 10
  
print ("Input  number:", in_num) 
   
    
out_arr = geek.atleast_2d(in_num) 
print ("output 2d array from input number:", out_arr)

輸出：

Input  number: 10
output 2d array from input number: [[10]]

代碼2：工作

# Python program explaining 
# numpy.atleast_2d() function 
  
import numpy as geek 
  
my_list = [2, 6, 10],  
   
print ("Input  list:", my_list) 
    
out_arr = geek.atleast_2d(my_list)  
print ("output 2d  array:", out_arr)

輸出：

Input  list: ([2, 6, 10], )
output 2d  array: [[ 2  6 10]]

代碼3：工作

# Python program explaining 
# numpy.atleast_2d() function 
# when inputs are in high dimension 
  
import numpy as geek 
  
in_arr = geek.arange(9).reshape(3, 3) 
print ("Input  array:\n ", in_arr) 
  
out_arr = geek.atleast_2d(in_arr) 
print ("output  array:\n ", out_arr) 
print(in_arr is out_arr)

輸出：

Input  array:
  [[0 1 2]
 [3 4 5]
 [6 7 8]]
output  array:
  [[0 1 2]
 [3 4 5]
 [6 7 8]]
True

相關用法

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