如何让一个np数组中随机几个数字+1

Question

比如我现在有[1,2,3,4,5,6]这个数组，

如何将该数组中随机n个数字都+1，

比如将原有数组变成[2,2,4,4,5,6],

这里面就随机到了第0第2个数加一了。

Answer 1

>>> import numpy as np
>>> x = np.array([1,2,3,4,5,6])
>>> r = np.random.choice(np.arange(x.size), 3, replace=False) # 从x中随机无重复地抽取3个数
>>> r # 这次选中的是0，2，4
array([0, 2, 4])
>>> x[r] += 1 # 选中的数字加1
>>> x
array([2, 2, 4, 4, 6, 6])