본문 바로가기

파이썬/파이썬 문법(심화)6

2차원 리스트를 회전시키기 2차원 리스트를 회전시켜야 할 때가 종종 있다.(특히 PS나, 코딩테스트에서 자주 등장한다.) 매우 난해한 문법이지만, 파이썬에서는 2차원 리스트를 단 한줄로 회전 시킬 수 있다. original = [[1, 2, 3], [4, 5, 6], [7, 8, 9]] # 시계방향 회전. rotated = list(zip(*original[::-1])) rotated2 = list(zip(*original))[::-1] print("시계방향 회전") print(*rotated, sep="\n") print("반시계방향 회전") print(*rotated2, sep="\n") >>> 시계방향 회전 (7, 4, 1) (8, 5, 2) (9, 6, 3) 반시계방향 회전 (3, 6, 9) (2, 5, 8) (1, 4, .. 2022. 10. 25.
0으로 초기화된 2-차원 리스트 생성하기. 프로그래밍 문제를 풀다 보면, 0으로 초기화된 2차원 리스트를 만들어야 할 때가 종종 있다. 여러가지 방법이 있겠지만, 안 좋은 예 2가지와 좋은 예 한가지를 소개한다. 예시1(좋음) #시간이 오래걸리는 경우 N = 10 ** 4 arr = [[0] * N for _ in range(N)] 예시2(시간이 느림) #시간이 오래걸리는 경우 N = 10 ** 4 arr = [[0 for _ in range(N)] for _ in range(N)] 예시3(예상하지 못한 결과가 발생함) #참조를 잘못한 예 N = 10 ** 4 arr = [[0] * N] * N 먼저 간단하게, time라이브러리를 이용하여 시간을 측정해보자. 예시1(좋음) #time 라이브러리 임포트 import time #생성 전 시간 변수저장.. 2022. 6. 28.

티스토리툴바