본문 바로가기
Programming/Python

[Python] input()함수 사용 시 if()문 사용하는 방법

by 고양이해적 2024. 4. 18.

 

input() 으로 입력을 받을때 if()문을 사용하여 입력데이터에 조건을 걸어야 할때

아래와 같이 한줄로 처리 할 수 있다.

 

 

📌 10 이하의 정수를 입력 받아, 공백으로 분리 후 정수형 리스트로 선언한다.

li = [int(x) for x in input().strip().split(' ') if int(x) <= 10]

 

 

 

📌 10 이하의 정수를 입력

#입력 예
1 2 3 4 5

#출력
[1, 2, 3, 4, 5]

 

 

📌 10 이상의 정수를 입력하면 리스트에 담지 않는다 

#입력 예
1 2 11

#출력
[1, 2]