파이썬 2차원 배열 input 쉽게 하기

dovelet문제를 풀다보면

2차원 배열에 대한 입력을 받는 문제들이 있다.

EX)

입력

1 2 3 4 5

1 2 3 4 5

2 3 4 5 2

1 2 3 4 5

4 2 1 2 3

다른 C언어라면 쉽게 배열에 데이터 값을 넣어줄 수 있지만 파이썬은 그렇지않다.

구글링을 해봐도 쉽게 한국어 자료를 찾을 수 없어서 간단하게 직접 만들어 보았다.

아주 간단하고 쉬운 소스지만 막상 오랜만에 파이썬을 하려니 기억이 안나실 분들을 위해서
2차원 배열 입력법을 다양하게 적어두었다.

num = int(input()) #세로열을 얼마나 줄지 정하는 부분

board =[[int(x) for x in input().split()]for y in range(num)] 데이터를 입력하는 부분

여기에 프린트를 추가시킨다면 실행결과는 아래와 같다. 

세로열을 지정을 해주었기에 5번 입력을 끝낼 수 있고 가로열을 split으로 스페이스단위로 끊었을 때 데이터를 나눈다.

2차원 배열 선언 간단히 하는 방법

num = int(input())

board = [[0]*num for y in range(num)]

선언 되어 있는 배열에 수를 입력

board[1][0] = 3 #매우 간단하게 그냥 입력 해주면 된다

이해를 돕기위해 사진을 준비했다.

num = int(input())

board = [[0]*num for x in range(num)]

print (board) # 데이터 삽입전

board[1][0] = 3

print (board) # 데이터 삽입 후

아직 더 많은 2차원 배열 입력 방법이 있겠지만 그것들은 대부분 이런 것들의 응용이기도 하고 찾다보면 너무 길어질거 같아 여기까지 적도록 하겠습니다. 그외에 궁금한 것들은 카톡으로 문의해주세요.