값 범위에 따라 컬럼 값 변경하는 pandas 예제

값 범위에 따라 컬럼 값을 변경하는 pandas 예제입니다.

2024. 3. 21  최초작성

label 컬럼에 있는 1에서 11 범위의 값을  주어진 조건에 따라 0과 1로 변경합니다. 

컬럼의 값이 1~5이면 1로, 6~11이면 0으로 변경합니다. 

 

import pandas as pd # 1에서 11 범위를 갖는 값을 가진 label 컬럼이 포함된 데이터프레임을 생성합니다. df = pd.DataFrame({     'label': [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11] }) # 컬럼에 포함된 값의 통계를 출력합니다. value_counts = df['label'].value_counts() print(value_counts) print('\n') # 'label' 컬럼의 값이 1~5이면 1로, 6~11이면 0으로 변경합니다. df.loc[df['label'].between(1, 5), 'label'] = 1 df.loc[df['label'].between(6, 11), 'label'] = 0 # 컬럼에 포함된 값의 통계를 다시 출력합니다. value_counts = df['label'].value_counts() print(value_counts)

실행결과입니다. label 컬럼에 1~11까지 숫자가 1개씩 있었는데 범위에 따라 값 변경후, 0은 6개, 1은 5개로 변경되었습니다.   

 

label

1     1

2     1

3     1

4     1

5     1

6     1

7     1

8     1

9     1

10    1

11    1

Name: count, dtype: int64

label

0    6

1    5

Name: count, dtype: int64