Skip to content Skip to sidebar Skip to footer

Replace Min Value To Another In Numpy Array

Lets say we have this array and I want to replace the minimum value with number 50 import numpy as np numbers = np.arange(20) numbers[numbers.min()] = 50 So the output is [50,1,2,

Solution 1:

You need to use numpy.argmin instead of numpy.min:

In [89]: numbers = np.arange(20).reshape(5,4)

In [90]: numbers[np.arange(len(numbers)), numbers.argmin(axis=1)] = 50
In [91]: numbers
Out[91]: 
array([[50,  1,  2,  3],
       [50,  5,  6,  7],
       [50,  9, 10, 11],
       [50, 13, 14, 15],
       [50, 17, 18, 19]])

In [92]: numbers = np.arange(20).reshape(5,4)

In [93]: numbers[1,3] = -5 # Let's make sure that mins are not on same column

In [94]: numbers[np.arange(len(numbers)), numbers.argmin(axis=1)] = 50

In [95]: numbers
Out[95]: 
array([[50,  1,  2,  3],
       [ 4,  5,  6, 50],
       [50,  9, 10, 11],
       [50, 13, 14, 15],
       [50, 17, 18, 19]])

(I believe my original answer was incorrect, I confused rows and columns, and this is right)


Post a Comment for "Replace Min Value To Another In Numpy Array"