python – 从一维数组的索引和值构造二维numpy数组

栏目: Python · 发布时间: 5年前

内容简介:翻译自:https://stackoverflow.com/questions/18968660/construct-two-dimensional-numpy-array-from-indices-and-values-of-a-one-dimension

说我有

Y = np.array([2, 0, 1, 1])

从这里我想获得一个形状矩阵X(len(Y),3).在这种特殊情况下,X的第一行应该在第二个索引上有一个,而在其他情况下为零. X的第二行应该在0索引上有一个,否则为零.要明确:

X = np.array([[0, 0, 1], [1, 0, 0], [0, 1, 0], [0, 1, 0]])

我该如何制作这个矩阵?

我开始了

X = np.zeros((Y.shape[0], 3))

但后来无法弄清楚如何填充/填写索引列表中的那些

一如既往,谢谢你的时间!

也许:
>>> Y = np.array([2, 0, 1, 1])
>>> X = np.zeros((len(Y), 3))
>>> X[np.arange(len(Y)), Y] = 1
>>> X
array([[ 0.,  0.,  1.],
       [ 1.,  0.,  0.],
       [ 0.,  1.,  0.],
       [ 0.,  1.,  0.]])

翻译自:https://stackoverflow.com/questions/18968660/construct-two-dimensional-numpy-array-from-indices-and-values-of-a-one-dimension


以上就是本文的全部内容,希望本文的内容对大家的学习或者工作能带来一定的帮助,也希望大家多多支持 码农网

查看所有标签

猜你喜欢:

本站部分资源来源于网络,本站转载出于传递更多信息之目的,版权归原作者或者来源机构所有,如转载稿涉及版权问题,请联系我们

Windows 程序设计:第5版

Windows 程序设计:第5版

CharlesPetzold / 北京博彦科技发展有限公司 / 北京大学出版社 / 2003-11-1 / 160.00元

Windows程序设计(第5版)对于Windows程序员来说,“从 Charles 的(Windows程序设计)一书中寻找答案。”几乎成了一句至理名言。而(Windows程序设计》第5版是专门为在Microsoft Windows 98、Microsoft Windows NT 4和 Windows NT 5下编程的开发人员编写的。内容博大精深,并有大量的源代码来帮助读者掌握Windows编程。本......一起来看看 《Windows 程序设计:第5版》 这本书的介绍吧!

JSON 在线解析
JSON 在线解析

在线 JSON 格式化工具

SHA 加密
SHA 加密

SHA 加密工具

RGB HSV 转换
RGB HSV 转换

RGB HSV 互转工具