Passage série avec DTYPE = « catégorie » comme catégories pour Pandas fonction catégorielles

Question

quand je lance ce code, je reçois l'erreur suivante:

import pandas as pd 

car_colors = pd.Series(['Blue', 'Red', 'Green'], 
      dtype='category') 

car_data = pd.Categorical(['Yellow', 'Green', 'Red', 'Blue','Purple'], 
           categories= car_colors, ordered=False) 
print car_colors 
s = pd.Series(car_data) 
s 

ValueError: object array method not producing an array

Mais le plus drôle est, quand je retire le dtype = 'category', le code fonctionne bien.

Donc en bref, la fonction catégorique est d'accepter la série mais pas avec dtype = 'category'

Est-ce un bug ou que je fais quelque chose de mal?

Answer 1

Il ressemble besoin d'ajouter tolist-categories dans Categorical:

car_colors = pd.Series(['Blue', 'Red', 'Green'], 
      dtype='category') 

car_data = pd.Categorical(['Yellow', 'Green', 'Red', 'Blue','Purple'], 
           categories = car_colors.tolist(), ordered=False) 

s = pd.Series(car_data) 
print (s) 

0  NaN 
1 Green 
2  Red 
3  Blue 
4  NaN 
dtype: category 
Categories (3, object): [Blue, Red, Green] 

Une autre solution de EdChum's comment est d'utiliser cat.categories:

car_data = pd.Categorical(['Yellow', 'Green', 'Red', 'Blue','Purple'], 
           categories = car_colors.cat.categories, ordered=False) 
s = pd.Series(car_data) 
print (s) 
0  NaN 
1 Green 
2  Red 
3  Blue 
4  NaN 
dtype: category 
Categories (3, object): [Blue, Green, Red]