본문 바로가기
5-3. Pandas 제공 자료구조/DataFrame

[파이썬 판다스] 데이터 프레임에서 특정 값을 기준으로 0과 1로 바꾸기

by 만다린망고 2021. 12. 30.
반응형

넘파이와 판다스를 불러옵니다. 

import numpy as np
import pandas as pd


배열을 하나 정의합니다. 

ar=np.array([[1,2,3],[4,5,6],[7,8,9],[10,11,12]])

>>> ar
array([[ 1,  2,  3],
       [ 4,  5,  6],
       [ 7,  8,  9],
       [10, 11, 12]])


배열을 이용하여 데이터프레임을 정의합니다.

df=pd.DataFrame(ar,columns=['C1','C2','C3'],index=['R1','R2','R3','R4'])

>>> df
    C1  C2  C3
R1   1   2   3
R2   4   5   6
R3   7   8   9
R4  10  11  12


5을 기준으로 5보다 같거나 큰 값은 1로, 5보다 작은 값은 0으로 바꿔봅시다. 

먼저 5보다 작은 값을 0으로 바꿉니다. 

>>> df[df<5]=0
>>> df
    C1  C2  C3
R1   0   0   0
R2   0   5   6
R3   7   8   9
R4  10  11  12


5보다 같거나 큰 값을 1로 바꿉니다 .

>>> df[df>=5]=1
>>> df
    C1  C2  C3
R1   0   0   0
R2   0   1   1
R3   1   1   1
R4   1   1   1

 

반응형

댓글