[파이썬] 조건문을 사용한 데이터 필터링

조건문은 프로그래밍에서 매우 중요한 개념이다. 특히 데이터 필터링과 관련하여 조건문은 매우 유용하게 사용된다. Python에서의 조건문을 사용하여 데이터를 필터링하는 방법을 알아보자.

1. 단순 조건문

단순 조건문은 가장 기본적인 조건문 형태이다. if 키워드를 사용하여 조건을 지정하고 그 조건이 참일 경우에만 실행되는 코드 블록을 작성한다.

number = 5

if number > 0:
    print("숫자는 양수입니다.")

위 예제에서는 number 변수가 0보다 큰지 확인하고, 참이라면 “숫자는 양수입니다.”라는 메시지를 출력한다.

2. 조건문과 반복문의 조합

조건문과 반복문을 조합하여 데이터를 필터링하는 경우도 많이 있다. 예를 들어, 리스트에서 특정 조건을 만족하는 값들만 추출하고자 할 때는 for 반복문과 함께 조건문을 사용한다.

numbers = [1, 2, 3, 4, 5]
even_numbers = []

for number in numbers:
    if number % 2 == 0:
        even_numbers.append(number)

print(even_numbers)

위 코드는 numbers 리스트에서 짝수인 값을 추출하여 even_numbers 리스트에 저장한 후, 해당 리스트를 출력한다.

3. 다중 조건문

다중 조건문은 하나 이상의 조건을 확인하여 실행할 코드 블록을 선택하는 방법이다. Python에서는 if, elif, else 키워드를 사용하여 다중 조건문을 작성한다.

score = 85

if score >= 90:
    grade = "A"
elif score >= 80:
    grade = "B"
elif score >= 70:
    grade = "C"
else:
    grade = "F"

print(grade)

위 예제에서는 score 변수에 저장된 값을 확인하여 학점을 결정한다. score가 90 이상이면 “A” 학점, 80 이상이면 “B” 학점, 70 이상이면 “C” 학점, 그 외의 경우에는 “F” 학점으로 결정된다.

4. 조건문의 응용 - 데이터 필터링

조건문을 사용하여 데이터를 필터링하는 예를 한번 살펴보자. 예를 들어, 특정 조건을 만족하는 학생들의 성적을 추출하는 경우를 가정해보자.

students = [
    {"name": "Alice", "score": 85},
    {"name": "Bob", "score": 90},
    {"name": "Charlie", "score": 70},
    {"name": "Dave", "score": 95}
]

high_score_students = []

for student in students:
    if student["score"] >= 90:
        high_score_students.append(student)

for student in high_score_students:
    print(student["name"], student["score"])

위 예제에서는 students 리스트에 각 학생의 정보를 딕셔너리 형태로 저장하고, 이를 순회하면서 성적이 90 이상인 학생들을 추출하여 high_score_students 리스트에 추가한다. 그 후, 해당 리스트의 학생 정보를 출력한다.

조건문을 사용한 데이터 필터링은 매우 유용하고 중요한 개념이다. Python의 조건문을 잘 활용하여 데이터를 원하는 조건에 맞게 추출하고 처리하는 능력을 갖추는 것은 데이터 분석 및 프로그래밍에 있어서 핵심적인 요소이다.