Eu tenho um conjunto de quadros de dados em que uma das colunas contém uma variável categórica. Gostaria de convertê-lo em várias variáveis fictícias, caso em que normalmente usariaget_dummies.

O que acontece é queget_dummies examina os dados disponíveis em cada quadro de dados para descobrir quantas categorias existem e, assim, criar o número apropriado de variáveis fictícias. No entanto, no problema em que estou trabalhando agora, eu realmente sei com antecedência quais são as categorias possíveis. Mas, ao analisar cada quadro de dados individualmente, nem todas as categorias necessariamente aparecem.

Minha pergunta é: existe uma maneira de passar paraget_dummies (ou uma função equivalente) os nomes das categorias, para que, para as categorias que não aparecem em um determinado quadro de dados, apenas crie uma coluna de 0s?

Algo que faria isso:

categories = ['a', 'b', 'c']

   cat
1   a
2   b
3   a

Torne-se este:

  cat_a  cat_b  cat_c
1   1      0      0
2   0      1      0
3   1      0      0