如何解决Numpy:为列表中的每个数组的 np.array 的每一行乘以 (1/2)^k
假设我有以下数组列表
dat = [np.array([[1,2],[3,4]]),np.array([[5,6]]),np.array([[1,[7,8],[2,3]]),4]])]
现在,对于列表中的每个元素,我想将数组的行乘以 (1/2)^k,其中 k 是每行的 (index + 1)。
以列表中的第一个和第三个数组为例
for np.array([[1,4]),it would becomes like
np.array([[0.5,1],[3/4,1])
for np.array([[1,[7/4,[1/4,3/8])
所以最终的结果应该是
newdat = [np.array([[0.5,1]]),np.array([[5/2,np.array([[1/2,3/8]]),np.array([[0.5,1]])]
对于我所做的只是使用 for 循环
for i in dat:
n = i.shape[0]
mul = (1/2)**np.arrange(1,n+1)
i = i*mul[:,None]
有没有更好的方法来做到这一点?
解决方法
不,不幸的是没有。由于您有一个具有不同形状的 np.arrays 列表,因此处理该列表的唯一方法是使用 for 循环。你可以使用列表理解,但它或多或少是一样的。 (它使用 for 循环)
dat = [e * (1/2)**np.arange(1,e.shape[0]+1)[:,None] for e in dat]
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。