[python] k, *s = list(map(int,input().split()))

Yewon Choi·2021년 1월 24일
1

Python

목록 보기
28/29

한 줄로 주어진 값 중
첫 번째 값은 k변수에 담고
그 뒤어 주어진 값은 s 리스트에 담는 방법

입력

7 1 2 3 4 5 6 7

공백을 사이로 한 줄로 여러 값이 입력된다.
이때, 첫 번재 값은 이후에 주어지는 값들의 개수와 같다.

k = 7
s = [1,2,3,4,5,6,7] (총 k개)

코드

k, *s = list(map(int,input().split()))
print(k, s)

출력

7 [1, 2, 3, 4, 5, 6, 7]

정리

data =list(map(int,input().split()))
k = data[0]
s = data[1:]

다음과 같이 코드 구현을 생각했었는데
한 줄로 더 간편하게 값을 입력받을 수 있는 것을 알게 되었다.
까먹을 거 같으니 적어둬야지.

profile
https://github.com/devAon 찰나의 개발흔적을 남기는 개발블로그 입니다 🐥 https://aonee.tistory.com 에서 Velog로 블로그 이전 작업중입니다 ! 🎶

0개의 댓글