python替换二维numpy数组中的值 [英] python replace values in 2d numpy array

问题描述

我想用-1代替2d numpy数组的每一列的最大值:

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

还有其他方法可以替换位置由maxposcol [i]给出的最大值吗?

Is there any other way to replace the max values whose positions are given by maxposcol[i]?

如果我想找到矩阵每一列的n个最大值，您会建议我怎么做?使用排序?反复重复搜索最大值，并在每一步将其替换?

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?

推荐答案

您可以这样做:

>>> a=np.argmax(b, axis=0)
>>> b[a] = -1
>>> b
array([[ 1,  2,  3,  4],
       [ 5,  6,  7,  8],
       [-1, -1, -1, -1]])

这篇关于python替换二维numpy数组中的值的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！