Python Pandas – 특정 요소가 Interval(구간)에 포함되는지 확인하는 방법

Pandas의 Interval 객체는 시작값과 끝값으로 정의된 연속적인 범위를 나타냅니다. 어떤 요소가 이 구간에 속하는지 확인하려면 파이썬의 in 연산자를 사용하면 간단하게 처리할 수 있습니다.

먼저 필요한 라이브러리를 임포트합니다.

import pandas as pd

Interval 생성하기

왼쪽 경계가 0, 오른쪽 경계가 10인 Interval을 생성해 보겠습니다.

interval = pd.Interval(left=0, right=10)

생성된 Interval을 출력하여 확인합니다.

print("Interval...\n", interval)

그다음 in 연산자를 사용해 특정 요소(여기서는 6)가 구간에 포함되는지 검사합니다.

print("\nThe specific element exists in the Interval? = \n", 6 in interval)

전체 예제 코드

아래는 위 과정을 모두 담은 전체 코드입니다.

import pandas as pd

# Interval 생성
interval = pd.Interval(left=0, right=10)

# Interval 출력
print("Interval...\n", interval)

# Interval 길이 출력
print("\nInterval length...\n", interval.length)

# 특정 요소가 Interval에 포함되는지 확인
print("\nThe specific element exists in the Interval? = \n", 6 in interval)

실행 결과

위 코드를 실행하면 다음과 같은 결과가 출력됩니다.

Interval...
(0, 10]

Interval length...
10

The specific element exists in the Interval? =
True

참고: Interval의 경계 포함 여부(closed)

출력 결과에서 볼 수 있듯이, 기본 설정으로 생성된 Interval은 (0, 10] 형태로 표현됩니다. 즉, 왼쪽 경계는 포함되지 않고 오른쪽 경계만 포함됩니다. 따라서 0 in interval은 False를 반환하고, 10 in interval은 True를 반환합니다.

경계 포함 여부는 closed 매개변수로 조절할 수 있으며, 'left', 'right', 'both', 'neither' 중 하나를 지정할 수 있습니다.

# 양쪽 경계를 모두 포함하는 Interval
interval = pd.Interval(left=0, right=10, closed='both')
print(0 in interval)   # True
print(10 in interval)  # True

이처럼 in 연산자와 closed 옵션을 함께 활용하면 다양한 조건의 구간 포함 여부를 유연하게 판별할 수 있습니다.