728x90
반응형
import numpy as np
n = int(input('n을 입력하시오 : ')) # 입력 받기
arr = np.zeros(n*n, dtype='int32') # 전부 0으로 채원진 배열 만들기
arr = arr.reshape(n,n) # 2차원 배열로 변환
for i in range(len(arr)): # 반복
arr[i][i] = 1 # ↘ 방향. arr[i][i]로 대각선 인덱스 찾고 해당 값을 1 부여
for i in range(len(arr)): # 반복
arr[i][len(arr)-1-i] = 1 # ↙방향. arr[i][i]로 대각선 인덱스 찾고 해당 값을 부여
print(arr)
728x90
반응형
'프로그램' 카테고리의 다른 글
[파이썬] 문제 : 학생 3일 달리기 기록 평균하여 내림차순 정렬 (0) | 2022.11.30 |
---|---|
[파이썬] 문제 : numpy 2차원 배열에서 원소의 전체합,행열 방향 합 구하기 (0) | 2022.11.30 |
[파이썬] 문제 : 7명의 학생 점수 입력 받아 합계,평균 계산 (0) | 2022.11.30 |
[파이썬] 문제 : 중첩 for문으로 숫자 삼각형 그리기 (1) | 2022.11.30 |
[파이썬] 문제 : numpy 2차원 배열 슬라이싱 (0) | 2022.11.30 |
댓글