python-特殊矢量和按元素乘法

内容导读

互联网集市收集整理的这篇技术教程文章主要介绍了python-特殊矢量和按元素乘法，小编现在分享给大家，供广大互联网技能从业者学习和参考。文章包含1003字，纯文字阅读大概需要2分钟。

内容图文

我有2个数组. “ A”是任意长度的一个(假设一个起点有1000个条目),其中每个点都具有一个n维向量,其中每个条目代表一个标量. “ B”是另一个,具有n个条目,每个条目包含一个3维向量.如何进行标量乘法,以使结果为一个数组“ C”,其中每个条目是n个标量中的每个标量与n个3维向量中的每个标量的乘积？

以4-D为例：

    a=[[1,2,3,4],[5,6,7,8],....]
    b=[[1,0,0],[0,1,0],[0,0,1],[1,1,1]]

结果

    c=[[1*[1,0,0],2*[0,1,0],3*[0,0,1],4*[1,1,1]] , [5*[1,0,0],...],...]

该实现应以numpy形式进行,而不要使用大的for循环,因为预计会有超过1000个条目.在我们的情况下,预期n为7.

解决方法:

如果您从以下内容开始：

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

然后,我们可以向a添加一个额外的轴,并沿着它重复数组就可以得到…

>>> a[:,:,None].repeat(3, axis=2)
array([[[1, 1, 1],
        [2, 2, 2],
        [3, 3, 3],
        [4, 4, 4]],

       [[5, 5, 5],
        [6, 6, 6],
        [7, 7, 7],
        [8, 8, 8]]])

现在,就像@Jaime所说的那样,在操作时无需使用重复,因为NumPy的广播会处理它：

>>> a[:,:,None] * b
array([[[1, 0, 0],
        [0, 2, 0],
        [0, 0, 3],
        [4, 4, 4]],

       [[5, 0, 0],
        [0, 6, 0],
        [0, 0, 7],
        [8, 8, 8]]])

内容总结

以上是互联网集市为您收集整理的python-特殊矢量和按元素乘法全部内容，希望文章能够帮你解决python-特殊矢量和按元素乘法所遇到的程序开发问题。 如果觉得互联网集市技术教程内容还不错，欢迎将互联网集市网站推荐给程序员好友。

内容备注

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

内容手机端