如何将每行的最大值和所有其他数字保持为0

Question

我希望保留每行的最大值，并将所有其他值更改为零， 像这样

a = np.array([0, 1, 2, 3, 4, 5, 6, 7, 9, 8])
should be 
[0,0,0,0,0,0,0,0,9,0]

到目前为止，我已经能够将最大值设置为1，将所有其他值设置为零

a = np.array([[0, 1, 2, 3, 4, 5, 6, 7, 9, 8]])
b = np.zeros_like(a)
b[np.arange(len(a)), a.argmax(1)] = 1
b

我怎样才能保留原值？ 我想把b乘以a，这可能会解决这个问题。但肯定有更简单的方法

Answer 1

试试这个：

b = np.zeros_like(a)
b[np.where(a==np.max(a))] = np.max(a)

看看您所做的尝试，我猜您已经错误地初始化了初始数组a，然后将argmax（）用作二维数组，如果数组a的初始化不正确，那么您的代码应该是这样工作的：

a = np.array([0, 1, 2, 3, 4, 5, 6, 7, 9, 8])

b = np.zeros_like(a)
b[np.argmax(a)] = np.max(a)