Python numpy.atleast_2d()
numpy.atleast_2d()函数用于当我们想将输入转换为至少有两个维度的数组时。标量和一维输入被转换为二维数组,而高维输入被保留。
语法 : numpy.atleast_2d(*arrays)
参数 :
arrays1, arrays2, … …。[array_like] 一个或多个类似数组的序列。非数组输入被转换为数组。已经有两个或更多维度的数组被保留下来。
返回 :
一个数组,或数组列表,每个数组都有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