글 읽기 - Python zip함수, zip object

hihunjin 3년 전 0

제 코드에서 matmult와 matmult2가 왜 다르게 나오나요?

zip을 할때와, zip을 안할때가 어떻게 다른지 모르겠어요.

3번째줄 zip_b의 output은 list of tuples로 나오긴 하는데요..

def matmult(a,b):
	zip_b = zip(*b)
	zip_b = list(zip_b)
	return [[sum(ele_a*ele_b for ele_a, ele_b in zip(row_a, col_b))%1000 for col_b in zip_b] for row_a in a]
 
def matmult2(a,b):
	zip_b = b
	return [[sum(ele_a*ele_b for ele_a, ele_b in zip(row_a, col_b))%1000 for col_b in zip_b] for row_a in a]
 
a, b =[[1, 0, 0, 0, 1], [1, 0, 0, 0, 1], [1, 0, 0, 0, 1], [1, 0, 0, 0, 1], [1, 0, 0, 0, 1]], [[1, 0, 0, 0, 1], [1, 0, 0, 0, 1], [1, 0, 0, 0, 1], [1, 0, 0, 0, 1], [1, 0, 0, 0, 1]]
print(matmult(a,b))
print(matmult2(a,b))
# print(matmult(a,b) == matmult2(a,b))        # False

seico75 3년 전 1

list(zip(*a))는 a행렬의 트랜스포즈를 구합니다.

>>> a
[[1, 2, 3], [4, 5, 6], [7, 8, 9]]
>>> list(zip(*a))
[(1, 4, 7), (2, 5, 8), (3, 6, 9)]
>>>

아시다시피 행렬의 곱셈은 a의 한행과 b의 한 열을 요소대요소로 곱한것의 합으로 행렬를 만들는 것이라서

a와 b의 트랜스포즈를 한 행씩 곱의 합을 구하는 방식으로 행렬의 곱을 구한 것이 matmult 함수입니다.

댓글을 작성하려면 로그인해야 합니다.

문제

문제

출처

ICPC

Python zip함수, zip object

hihunjin 3년 전 0

seico75 3년 전 1

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말