[파이썬] 문제 : 배열에서 전부 0으로 채운후 경계 값만 1로 만들기

import numpy as np

n = int(input('n을 입력하시오 : '))

a = np.ones((n,n), dtype='int32')
b = a.copy()            # 복사
b[1:-1, 1:-1] = 0       # 내부를 0 으로 만듬

print('a 행렬\n',a)
print('b 행렬\n',b)

 

 

import numpy as np

n = int(input('n을 입력하시오 : '))

a = np.ones((n,n), dtype='int32')
c = a.copy()          # 복사
c[0,:] = 0            # 위
c[-1,:] = 0          # 아래
c[:,0] = 0           # 왼쪽
c[:,-1] = 0          # 오른쪽


print('a 행렬\n',a)
print('c 행렬\n',c)

또는

import numpy as np

n = int(input('n을 입력하시오 : '))

a = np.ones((n,n), dtype='int32')
c = a.copy()          # 복사
c[0:,0:] = 0          # 전부 0으로 만듬
c[1:-1, 1:-1] = 1     # 내부를 1로 만듬

print('a 행렬\n',a)
print('c 행렬\n',c)