Передача серии с DTYPE = «категории» как категории для панды категоричных функции

Question

, когда я запускаю этот код, я получаю следующее сообщение об ошибке:

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

Но самое смешное, когда я удалить dtype = 'category', код работает нормально.

Короче говоря, категорический функция принимает серию, но не с dtype = 'category'

Является ли это ошибка или я делаю что-то не так?

Answer 1

Похоже, нужно добавить tolist к categories в 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] 

Другим решением от EdChum's comment является использование 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]