python – Numpy：将每行的最大值更改为1,将所有其他数字更改为0

内容导读

互联网集市收集整理的这篇技术教程文章主要介绍了python – Numpy：将每行的最大值更改为1,将所有其他数字更改为0，小编现在分享给大家，供广大互联网技能从业者学习和参考。文章包含872字，纯文字阅读大概需要2分钟。

内容图文

我正在尝试实现一个numpy函数,它将2D数组的每一行中的max替换为1,并将所有其他数字替换为0：

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

到目前为止我尝试过的

def some_function(x):
    a = np.zeros(x.shape)
    a[:,np.argmax(x, axis=1)] = 1
    return a

>>> b = some_function(a)
>>> b
[[1. 1.]
 [1. 1.]
 [1. 1.]
 [1. 1.]
 [1. 1.]]

解决方法:

方法#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
array([[0, 1],
       [0, 1],
       [0, 1],
       [0, 1],
       [1, 0]])

[实际上,范围会很好;我出于习惯写了一个范围.

方法#2,使用max而不是argmax来处理多个元素达到最大值的情况：

>>> a = np.array([[0, 1], [2, 2], [4, 3]])
>>> (a == a.max(axis=1)[:,None]).astype(int)
array([[0, 1],
       [1, 1],
       [1, 0]])

内容总结

以上是互联网集市为您收集整理的python – Numpy：将每行的最大值更改为1,将所有其他数字更改为0全部内容，希望文章能够帮你解决python – Numpy：将每行的最大值更改为1,将所有其他数字更改为0所遇到的程序开发问题。 如果觉得互联网集市技术教程内容还不错，欢迎将互联网集市网站推荐给程序员好友。

内容备注

版权声明：本文内容由互联网用户自发贡献，该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容， 请发送邮件至 gblab@vip.qq.com 举报，一经查实，本站将立刻删除。

内容手机端