I want to replace the max values of each column of a 2d numpy array with -1:
b = numpy.array([[1,2,3,4],[5,6,7,8], [9,10,11,12]])
#get the max value of each column
maxposcol = b.argmax(axis = 0)
maxvalcol = b.max(axis = 0)
#replace max values with -1
for i in numpy.arange(b.shape[1]):
b[maxposcol[i]][i] = -1
Is there any other way to replace the max values whose positions are given by maxposcol[i]?
If I want to find the n maximal values of each column of my matrix what would you advise me to do? Using a sort? Repeating iteratively the search of max value and replace them at each step?
解决方案
You can do it like this:
>>> a=np.argmax(b, axis=0)
>>> b[a] = -1
>>> b
array([[ 1, 2, 3, 4],
[ 5, 6, 7, 8],
[-1, -1, -1, -1]])