quinta-feira, 15 de novembro de 2018

# Autor: alexciuffa
import pandas as pd

d = {'PSG':['AAA', 'BBB', 'CCC', 'DDD'], 'CLASS':[1,2,3,1]}
df = pd.DataFrame(data=d)

>>> print df CLASS PSG 0 1 AAA 1 2 BBB 2 3 CCC 3 1 DDD
# Criamos uma regra para definir as classes def define_classe(num): if num == 1: return 'PRIMEIRO' elif num == 2: return 'SEGUNDO' elif num == 3: return 'TERCEIRO' return 'SEM_CLASSE' # E ai aplicamos a regra df['CLASSE'] = df['CLASS'].map(define_classe)
>>> print df CLASS PSG CLASSE 0 1 AAA PRIMEIRO 1 2 BBB SEGUNDO 2 3 CCC TERCEIRO 3 1 DDD PRIMEIRO

0 comentários: