如何解决有没有办法在Python中将多维数组重塑为一维向量？
所以我想把这个数组转
array([[1.,0.,0.],[0.,1.,1.],0.]])
[1.,0.]])
进入此数组
array([ 0,4,9,6,0])
因此，将原始数组的每一行替换为一个等于该行中“ 1”所在位置的值。
解决方法
您可以使用numpy.argmax：
a = np.array([[1.,0.,0.],[0.,1.,1.],[1.,0.]])
print( np.argmax(a,axis=1) )
打印：
[0 3 9 6 0]
尝试一下：
array = np.array([[1.,0.]])
print(np.where(array == 1)[1]
输出：
array([0,3,9,6,0])
版权声明：本文内容由互联网用户自发贡献，该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容， 请发送邮件至 dio@foxmail.com 举报，一经查实，本站将立刻删除。
