python替换2d numpy数组中的值

时间:2012-06-11 17:26:13

标签: python multidimensional-array max

我想用-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个最大值,你会建议我做什么?使用排序?迭代地重复搜索最大值并在每一步替换它们?

1 个答案:

答案 0 :(得分:3)

你可以这样做:

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