2

2D numpy 配列の各列の最大値を -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] で指定された位置の最大値を置き換える他の方法はありますか?

行列の各列の n 個の最大値を見つけたい場合、どうすればよいでしょうか? ソートを使用していますか？最大値の検索を繰り返し繰り返し、各ステップでそれらを置き換えますか?

4

1 に答える 1

3

次のように実行できます。

>>> a=np.argmax(b, axis=0)
>>> b[a] = -1
>>> b
array([[ 1,  2,  3,  4],
       [ 5,  6,  7,  8],
       [-1, -1, -1, -1]])
于 2012-06-11T17:47:30.623 に答える