Olá Davi,

Basicamente o arquivo .py precisa estar na mesma pasta onde está salvo o arquivo do Jupyter que você está trabalhando agora. Veja abaixo mais sobre o assunto

https://stackoverflow.com/questions/56481126/how-to-use-code-from-external-python-files-in-jupyter-notebook

Jones