python-以numpy返回矩阵的邻居之和的最简单方法

内容导读

互联网集市收集整理的这篇技术教程文章主要介绍了python-以numpy返回矩阵的邻居之和的最简单方法，小编现在分享给大家，供广大互联网技能从业者学习和参考。文章包含1270字，纯文字阅读大概需要2分钟。

内容图文

我正在尝试制作一个需要矩阵的邻居(不包括自身)之和的程序ex ex：

 matrix([[0, 0, 0],
        [1, 0, 1],
        [0, 1, 0]])  

会返回：

matrix([[1, 2, 1],
        [1, 3, 1],
        [2, 2, 2]])

我在这里有一个有效的代码,但是它又大又杂乱,我是numpy的新手,所以我需要一些帮助来清理和优化它. (我觉得必须有更好的方法)

示例代码：

import numpy as np

def NiSum(m):
    new = []
    for x in range(m.shape[0]-1):
        row = []
        for y in range(m.shape[1]-1):
            Ni = 0
            for a in [1,1],[1,0],[1,-1],[0,1],[0,-1],[-1,1],[-1,0],[-1,-1]:
                Ni += m[x+a[0],y+a[1]]
            row.append(Ni)
        new.append(row)
    return np.matrix(new)


example = np.matrix('0 0 0 0 0 0 0 0; '*3+'0 0 0 1 1 1 0 0; '*3+'0 0 0 0 0 0 0 0;0 0 0 0 0 0 0 0 ')

NiSum(example)

谢谢你的帮助 ！

解决方法:

您正在对该3×3邻域中的所有值求和,但不包括元素本身.因此,我们可以使用Scipy's 2D convolution并从中减去该输入数组/矩阵以获得所需的输出,如下所示：

from scipy.signal import convolve2d

convolve2d(a,np.ones((3,3),dtype=int),'same') - a

样品运行-

In [11]: a
Out[11]: 
matrix([[0, 0, 0],
        [1, 0, 1],
        [0, 1, 0]])

In [12]: convolve2d(a,np.ones((3,3),dtype=int),'same') - a
Out[12]: 
matrix([[1, 2, 1],
        [1, 3, 1],
        [2, 2, 2]])

或简单地形成一个全为零但中心为零的内核,并使用相同的2D卷积-

In [31]: kernel = np.array([[1,1,1],[1,0,1],[1,1,1]])

In [32]: np.asmatrix(convolve2d(a,kernel,'same'))
Out[32]: 
matrix([[1, 2, 1],
        [1, 3, 1],
        [2, 2, 2]])

内容总结

以上是互联网集市为您收集整理的python-以numpy返回矩阵的邻居之和的最简单方法全部内容，希望文章能够帮你解决python-以numpy返回矩阵的邻居之和的最简单方法所遇到的程序开发问题。 如果觉得互联网集市技术教程内容还不错，欢迎将互联网集市网站推荐给程序员好友。

内容备注

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

内容手机端