Python中的 numpy.atleast_2d()

numpy.atleast_2d()函数用于将输入转换为至少具有二维的数组。标量和一维输入被转换为二维数组，而高维输入被保留。

Syntax : numpy.atleast_2d(*arrays)

Parameters :
arrays1, arrays2, … : [array_like] One or more array-like sequences. Non-array inputs are converted to arrays. Arrays that already have two or more dimensions are preserved.

Return : An array, or list of arrays, each with a.ndim >= 2. Copies are avoided where possible, and views with two or more dimensions are returned.

代码#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