Get mapowania zmiennych kategorycznych w pand

Question

Robię to, aby skategoryzowanych numery

>>> df = pd.DataFrame({'x':['good', 'bad', 'good', 'great']}, dtype='category') 

     x 
0 good 
1 bad 
2 good 
3 great 

Jak mogę uzyskać odwzorowanie między oryginalnymi wartościami i nowymi wartościami?

Answer 1

Można utworzyć słownika mapowanie przez wyliczanie (podobny do tworzenia słownika z listy poprzez tworzenie kluczy słownik z indeksów lista):

>>> dict(enumerate(df['x'].cat.categories)) 

{0: 'bad', 1: 'good', 2: 'great'} 

aby sprawdzić, czy to działa, po prostu wydrukować bazowego całkowitej kody takie jak ten:

>>> df['x'].cat.codes 

0 1 
1 0 
2 1 
3 2 
dtype: int8 

czy goście mogą skorzystać z kategorii, do liczby całkowitej na ciąg do kategorii:

>>> df['x'].cat.codes.map(dict(enumerate(df['x'].cat.categories))).astype('category') 

uzyskać ogólne informacje na pand zmienne kategoryczne widzą official documentation