Python|Numpy加粗

問題描述

示列

輸入：

[[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. 1. 1. 1. 0. 0.]

 [0.0. 1. 1. 1. 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