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 團隊

---

   

微信號：算法與編程之美          

長按識別二維碼關注我們!

溫馨提示：點擊頁面右下角“寫留言”發表評論，期待您的參與！期待您的轉發！