In Python, the numpy.ones_like()
function is used to return an array of ones, 1
, with the same shape as a given array.
numpy.ones_like(a, dtype=None, order='K')
The numpy.ones_like()
function takes the following parameter values:
a
: This represents the array_like
object or the prototype.dtype
: This represents the data type of the result. This parameter is optional.order
: This represents the type of order of the memory layout of the result. This is an optional parameter.The numpy.ones_like()
function returns an array of ones, 1
, with the same shape and types as the array_like
that is passed to it.
import numpy as np# creating an array_likethisarray = np.arange(4, dtype = int)# implementing the numpy.ones_like() functionmyarray = np.ones_like(thisarray)# printing the two arraysprint(thisarray)print(myarray)
Line 1: We import the numpy
module.
Line 4: We create an array prototype, using the numpy.arange()
method. We assign the output to a variable called thisarray
.
Line 7: We implement the numpy.ones_like()
function on thisarray
and assign the result to another variable called myarray
.
Lines 11–12: We print both the prototype thisarray
and the modified array myarray
.