[백준/python] 2738번 행렬 덧셈: 행렬의 크기와 원소 받아 2차원 배열 만들기

 

n,m=map(int, input().split())

A=[0]*n
B=[0]*n

for i in range(n):
        A[i]=list(map(int,input().split()))

for i in range(n):
        B[i]=list(map(int,input().split()))

for i in range(n):
        for j in range(m):
                print(A[i][j]+B[i][j], end=' ')
        print()

 

 

A=[0]*n으로 초기화해 (n,m=3이라고 하면 A=[0, 0, 0]으로 초기화됨)

그리고 받은거를 리스트화해 i번째에 넣어줌 즉

1 1 1

2 2 2

3 3 3

이런식으로 입력받으면, A[[1,1,1], [2,2,2], [3,3,3]] 이렇게 됨

m=4이고 입력이

1 1 1 1

2 2 2 2

3 3 3 3

이런식으로 들어와도 A[[1,1,1,1], [2,2,2,2], [3,3,3,3]] 이렇게 됨

 

이 방법 말고도 그냥 행단위로 append해주는 방법도 있음

A=[]
for i in range(n):
    A.append(list(map(int, input().split())))