[백준] 2738번 파이썬

2738번 - 행렬 덧셈 

아래 코드는 백준에서 쓸 수 있는 코드이다.

import sys

n, m = map(int,sys.stdin.readline().split())
mat_1 = []
mat_2 = []
for i in range(n):
    row = list(map(int,sys.stdin.readline().split()))
    mat_1.append(row)

for i in range(n):
    row = list(map(int,sys.stdin.readline().split()))
    mat_2.append(row)

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

---

아래 코드는 백준에서는 쓸 수 없다. 백준에서는 pandas 라이브러리를 사용할 수 없다.

import sys
from pandas import DataFrame

n, m = map(int,sys.stdin.readline().split())
mat_1 = []
mat_2 = []
for i in range(n):
    row = list(map(int,sys.stdin.readline().split()))
    mat_1.append(row)
    df1 = DataFrame(mat_1)

for i in range(n):
    row = list(map(int,sys.stdin.readline().split()))
    mat_2.append(row)
    df2 = DataFrame(mat_2)
df = df1.add(df2)

for i in range(n):
    r = df.loc[i]
    for j in range(m):
        print(r[j], end=" ")
    print()

마지막의 프린트를 이용하면 줄바꿈을 할 수 있다.