python数组替换,python替换二维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

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

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值