Olá Hiago, tudo bem com você?
Para corrigir o erro de ModuleNotFoundError
tente executar o comando de instalação da biblioteca pyspark
dentro do próprio Jupyter Notebook. Crie duas novas célula e execute:
!pip install pyspark
!pip install findspark
Posteriormente a instalação, tente executar outra vez os códigos abaixo:
import findspark
findspark.init()
from pyspark.sql import SparkSession
Espero ter ajudado. Continue mergulhando em conhecimento e não hesite em voltar ao fórum para continuar aprendendo e interagindo com a comunidade. Em caso de dúvidas estou à disposição. Abraços e bons estudos!