파이썬(python) 코테 연습용 파일 읽는방법

항상 자바 스크립트로만 코딩테스트를 보다가 최근 파이썬으로 언어를 바꿨다.

 

이유는

1. 자료 및 문제의 총량 차이

2. 문법에서 오는 간결함과 그로인해 얻어지는 시간적이다

이다.

 

특히나 vscode에서 디버깅을 통해서 변수값을 알수있는것도 마이너하지만 확실한 이득이라고 생각한다.

백준등의 문제에서는 input으로 입력을 받는경우가 많다.
이때, vscode에서 테스트케이스를 테스트할때 터미널에서 매번 타이핑해야하는것이 귀찮았다.

물론, 테스트케이스를 다른데 복사한다음에 ctrl c + v하면 되긴하겠지만 그런 원시적인 방법대신
나는 txt 파일을 읽게 함으로서 테스트케이스를 적용하고 싶었다

 

import sys
sys.stdin = open(
    '~~~~~/iceMaker.txt', 'r')

temp = list(map(int, input().split(' ')))
N = temp[0]
M = temp[1]

BLOCK = []
for i in range(0, N):
    BLOCK.append(input().split())

iceMaker.py

4 5
00110
00011
11111
00000

iceMaker.txt

 

위의 코드가 iceMaker.txt를 읽어들이는 소스다.

open을 한다음에 readline을 하는 방법도 있지만, 나는 입력으로 하고 싶어서 stdin함수를 썻다.

위와 같이 파일로 테스트 코드를 만들어서 사용하면 매번 복사 붙여넣기 할 필요가 없다.