您好, 欢迎来到 !    登录 | 注册 | | 设为首页 | 收藏本站

Numpy,python:广播时自动扩展阵列的尺寸

5b51 2022/1/14 8:22:49 python 字数 1508 阅读 631 来源 www.jb51.cc/python

在Numpy阵列广播中考虑以下练习. import numpy as np v = np.array([[1.0, 2.0]]).T # column array A2 = np.random.randn(2,10) # 2D array A3 = np.random.randn(2,10,10) # 3D v * A2 # works great # causes error: v *

概述

import numpy as np
v = np.array([[1.0,2.0]]).T # column array

A2 = np.random.randn(2,10) # 2D array
A3 = np.random.randn(2,10,10) # 3D

v * A2 # works great

# causes error: 
v * A3 # error

我知道广播的Numpy规则,我熟悉Matlab中的bsxfun功能.我明白为什么试图将(2,1)数组广播到(2,N,N)数组中失败,我必须在(2,1)数组之前将(2,1)数组重新形成广播经过.

我的问题是:有没有办法告诉Python在尝试广播时自动填充数组的维度,而我不必专门告诉它必要的维度?

我不想明确地将(2,1)向量与要广播的多维数组对齐 – 否则我可以做一些像mult_v_A = lambda v,A:v.reshape([ v.size] [1] *(A.ndim-1))* A.如果“A”数组将是2D或3D或ND,我不知道如何.

Matlab的bsxfun广播功能根据需要隐含地填充维度,所以我希望有一些我可以在Python中做的事情.

(v.T * A3.T).T

如果你不给它任何论据,转置形状元组,所以你现在可以依靠广播规则做他们的魔法.最后一个转置返回所有的顺序.

总结

以上是编程之家为你收集整理的Numpy,python:广播时自动扩展阵列的尺寸全部内容,希望文章能够帮你解决Numpy,python:广播时自动扩展阵列的尺寸所遇到的程序开发问题。


如果您也喜欢它,动动您的小指点个赞吧

除非注明,文章均由 laddyq.com 整理发布,欢迎转载。

转载请注明:
链接:http://laddyq.com
来源:laddyq.com
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。


联系我
置顶