numpy.atleast_2d () in Python



numpy.atleast_2d() is used when we want to convert the input to arrays with at least two dimensions. Scalar and one-dimensional inputs are converted to 2-dimensional arrays, while larger inputs are preserved.

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 & gt; = 2. Copies are avoided where possible , and views with two or more dimensions are returned.

Code # 1: Work

# 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) 

Output:

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

Code # 2: Work

# 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) 

Output:

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

Code # 3: Work

# 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:" , in_arr )

 

out_arr = geek.atleast_2d (in_arr)

print ( "output array:" , out_arr)

print (in_arr is out_arr)

Output:

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