Oi Gilmar, como vai?
Peço perdão pela demora em obter um retorno.
Talvez tenha alguma variável tenha sido sobrescrita em algum momento. Entretanto, testei o código que você apresentou e tudo ocorreu como esperado.
Vamos fazer alguns testes:
- Abra uma nova célula de código no Jupyter(como mostro na imagem abaixo), cole e tente executar o código:

- Faça um restart das variáveis deste notebook. Basta ir em Kernel > Restart e após isso, tente executar o código novamente :

Me informe se o erro persiste.
Qualquer dúvida estou por aqui.
Abraços e bons estudos!