Oii Felipe, como você está?
Desculpe pela demora em te responder.
Seu código está certinho. O erro pode estar vindo caso pd
ou DataFrame
estejam como variáveis em algum lugar anterior ao seu código. Mas, o provável é que seu ambiente precise ser reiniciado. Para fazer isso, vá em ambiente de execução, no menu superior, e selecione reiniciar e executar tudo, ou apenas em reiniciar ambiente de execução, caso você não deseje que todas as células sejam executadas.
Espero que dê tudo certo por aí. Fico à disposição.
Abraços!
Caso este post tenha lhe ajudado, por favor, marcar como solucionado ✓. Bons Estudos!