[파이썬] 삼항연산자

프로그래머스 기초 단계의 파이썬 문제풀이 후 학습한 내용입니다.

---

문제 설명

홀짝 구분하기

자연수 n이 입력으로 주어졌을 때 만약 n이 짝수이면 "n is even"을,
홀수이면 "n is odd"를 출력하는 코드를 작성해 보세요.

기존 코드

import sys
input = sys.stdin.readline

a = int(input())

res = ' is even' if a%2==0 else ' is odd'
print(f'{a}'+ res)

---

삼항 연산자

삼항 연산자

• 간단한 조건의 경우, if문을 사용하는 것보다 코드를 간결하게 나타낼 수 있음

• 기본 문법

  • if문만 단독으로 사용하거나, if-else문을 완전한 형태로 사용해야 함
  1. if문만 단독 사용
     • if문 단독으로 사용할 때 if문이 for문 앞에 오면 안됨

		[x for x in 리스트 if 조건]

2. if-else문 사용

		[x if 조건 else y for x in 리스트]

• 예시

# 일반 if-else문
if score >= 60:
    result = "Pass"
else:
    result = "Fail"

# 삼항 연산자
result = "Pass" if score >= 60 else "Fail"

f-string 내부에 삼항 연산자 사용

• 기본 사용법 예시

age = 20
print(f"당신은 {'성인' if age >= 18 else '미성년자'}입니다")

• 중괄호 안에서 삼항 연산자를 사용할 때는 전체를 작은따옴표나 큰 따옴표로 묶어야 함
  • f-string 내에서 삼항 연산자를 사용할 때는 바깥쪽과 안쪽의 따옴표를 다르게 사용해야 함

# 올바른 사용
print(f"결과: {'성공' if score > 80 else '실패'}")

# 잘못된 사용
print(f"결과: {성공 if score > 80 else 실패}")  # 에러 발생

---

코드 개선

f-string 내부에 삼항 연산자를 사용한 간략한 코드

import sys
input = sys.stdin.readline

a = int(input())

print(f'{a} is {"even" if a%2==0 else "odd"}')