蟒蛇代替值在2D numpy的陣列

Question

我想用-1替換2D numpy的陣列的每列的最大值：

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]？

如果我想找到矩陣每列的n個最大值，你會建議我做什麼？使用排序？重複迭代搜索最大值並在每一步中替換它們？

Answer 1

你可以這樣說：

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