본문 바로가기
Algorithm

파이썬 문법 - 입출력

by Kwonys 2024. 4. 1.
728x90
반응형

1. 입력

1-1. 자주 사용되는 표준 입력 방법

  • input( ) : 한 줄의 문자열을 입력 받는 함수
  • map( ) : 리스트의 모든 원소에 각각 특정한 함수를 적용할 때 사용
  • ex) 공백을 기준으로 구분된 데이터를 입력 받을 때
    • list(map(int, input().split()))
  • ex) 공백을 기준으로 구분된 데이터의 개수가 많지 않을 때
    • a, b, c = map(int, input().split())

1-2. 빠르게 입력 받기

정수를 받는 경우

import sys
a = int(sys.stdin.readline())

a1,a2,a3 = map(int,sys.stdin.readline().split())

data = list(map(int,sys.stdin.readline().split()))

> 1 2 3 4

print(data)

>[1,2,3,4]

문자열 n줄을 입력 받아 리스트에 저장하는 경우

내가 sys.stdin.readline()을 사용하면서 가장 많은 에러를 발생하였던 장소로
이유를 모르고 input()으로 고치거나 갑자기 되네?? 가 가장 많았던 오류였다.

import sys
cnt = int(sys.stdin.readline())
data = [sys.stdin.readline().strip() for i in range(cnt)]

print(data)

> 2
> data1
> data2

>>['data1', 'data2']

 

strip()을 통해서 개행문자를 제거하지 않은 경우

> 2
> data1
> data2

>>['data1\n', 'data2\n']
  • 이진탐색, 정렬, 그래프 관련 문제에서 자주 사용

2. 출력

  • print()는 기본적으로 출력 이후 줄 바꿈 수행
  • 줄 바꿈을 원치 않는 경우 'end' 속성 이용
    • ex)
print(7, end= " ")
print(8)
# 수행 결과 : 7, 8
728x90
반응형