python – 添加不同形状的numpy数组

内容导读

互联网集市收集整理的这篇技术教程文章主要介绍了python – 添加不同形状的numpy数组，小编现在分享给大家，供广大互联网技能从业者学习和参考。文章包含1060字，纯文字阅读大概需要2分钟。

内容图文

我想添加两个不同形状的numpy数组,但没有广播,而是将“缺失”值视为零.像一个例子可能最简单

[1, 2, 3] + [2] -> [3, 2, 3]

要么

[1, 2, 3] + [[2], [1]] -> [[3, 2, 3], [1, 0, 0]]

我事先不知道形状.

我正在弄乱每个np.shape的输出,试图找到包含它们的最小形状,将每个形状嵌入该形状的零编辑数组然后添加它们.但似乎有很多工作,是否有更简单的方法？

提前致谢！

编辑：通过“很多工作”我的意思是“为我做了很多工作”而不是机器,我寻求优雅而不是效率：我努力获得最小的形状同时保持它们两者是

def pad(a, b) :
    sa, sb = map(np.shape, [a, b])
    N = np.max([len(sa),len(sb)])
    sap, sbp = map(lambda x : x + (1,)*(N-len(x)), [sa, sb])
    sp = np.amax( np.array([ tuple(sap), tuple(sbp) ]), 1)

不漂亮 ：-/

解决方法:

这是我能想到的最好的：

import numpy as np

def magic_add(*args):
    n = max(a.ndim for a in args)
    args = [a.reshape((n - a.ndim)*(1,) + a.shape) for a in args]
    shape = np.max([a.shape for a in args], 0)
    result = np.zeros(shape)

    for a in args:
        idx = tuple(slice(i) for i in a.shape)
        result[idx] += a
    return result

如果您知道对结果有多少维度,可以稍微清理for循环,例如：

for a in args:
    i, j = a.shape
    result[:i, :j] += a

内容总结

以上是互联网集市为您收集整理的python – 添加不同形状的numpy数组全部内容，希望文章能够帮你解决python – 添加不同形状的numpy数组所遇到的程序开发问题。 如果觉得互联网集市技术教程内容还不错，欢迎将互联网集市网站推荐给程序员好友。

内容备注

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

内容手机端