Pandas: mapeamento de tuplas em DataFrame

# 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