Python|Numpy加粗

欢迎点击「算法与编程之美」↑关注我们！

本文首发于微信公众号："算法与编程之美"，欢迎关注，及时了解更多此系列文章。

问题描述

示列

输入：

[[0. 0. 0. 0. 0. 1. 0. 0.]

 [0.0. 0. 0. 0. 1. 0. 0.]

 [0.0. 0. 0. 0. 1. 0. 0.]

 [0.0. 0. 0. 1. 0. 0. 0.]

 [0.0. 0. 1. 0. 0. 0. 0.]

 [0.0. 1. 0. 0. 0. 0. 0.]

 [1.1. 1. 0. 0. 0. 0. 0.]

 [0.0. 0. 0. 0. 0. 0. 0.]]

输出：

[[0. 0. 0. 0. 1. 1. 1. 0.]

 [0.0. 0. 0. 1. 1. 1. 0.]

 [0.0. 0. 0. 1. 1. 1. 0.]

 [0.0. 0. 1. 1. 1. 0. 0.]

 [0.0. 1. 1. 1. 0. 0. 0.]

 [1.1. 1. 1. 0. 0. 0. 0.]

 [1.1. 1. 1. 0. 0. 0. 0.]

 [1.1. 1. 0. 0. 0. 0. 0.]]

解决方案

加粗可以理解为初始数组向4个方向（上，下，左，右）平移后在取并集得到的。

Python代码：

import numpy as np

arr=np.zeros((8,8))

arr[0:3,5]=1                    

arr[3,4]=1

arr[4,3]=1

arr[5,2]=1

arr[6,0:3]=1                    #生成初始数组

#print(arr)

def jiacu(arr):

     arr1=np.roll(arr,1,axis=1)

     arr1[0:7,0]=0               #向右平移

     arr2=np.roll(arr,-1,axis=1)

     arr2[0:7,7]=0               #向左平移

     arr3=np.roll(arr,1,axis=0)

     arr3[0,0:7]=0               #向下平移

     arr4=np.roll(arr,-1,axis=0)

     arr4[7,0:7]=0               #向上平移

     arr5=arr1+arr2+arr3+arr4+arr

     for i in range(len(arr)):

         for j in range(len(arr[0])):

            if arr5[i][j]>1:

                arr5[i][j]=1    #取并集

     return arr5

print(jiacu(arr))

END

主       编   |   王楠岚

责       编   |   王自强

 where2go 团队

---

   

微信号：算法与编程之美          

长按识别二维码关注我们!

温馨提示：点击页面右下角“写留言”发表评论，期待您的参与！期待您的转发！