问题描述
示列
输入:
[[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 团队