Numpy ndarray.itemset() function | Python
numpy.ndarray.itemset() function insert scalar into an array.
There must be at least 1 argument, and define the last argument as item. Then, arr.itemset(*args) is equivalent to but faster than arr[args] = item. The item should be a scalar value and args must select a single item in the array arr.
Syntax : numpy.ndarray.itemset(*args)
Parameters :
*args : If one argument: a scalar, only used in case arr is of size 1. If two arguments: the last argument is the value to be set and must be a scalar, the first argument specifies a single array element location. It is either an int or a tuple.
Code #1 :
# Python program explaining # numpy.ndarray.itemset() function # importing numpy as geek import numpy as geek geek.random.seed(345) arr = geek.random.randint(9, size =(3, 3)) print("Input array : ", arr) arr.itemset(4, 0) print ("Output array : ", arr) |
Output :
Input array : [[8 0 3] [8 4 3] [4 1 7]] Output array : [[8 0 3] [8 0 3] [4 1 7]]
Code #2 :
# Python program explaining # numpy.ndarray.itemset() function # importing numpy as geek import numpy as geek geek.random.seed(345) arr = geek.random.randint(9, size =(3, 3)) print("Input array : ", arr) arr.itemset((2, 2), 9) print ("Output array : ", arr) |
Output :
Input array : [[8 0 3] [8 4 3] [4 1 7]] Output array : [[8 0 3] [8 4 3] [4 1 9]]
