[D] Best software to write a machine learning based master thesis • r/MachineLearning

I am studying a MSc in Applied Statistics and my master thesis is called "Unsupervised learning techniques applied to classification of gymnasts through the measuring of individual elements from acrobatic gymnastic discipline". That is, I would need to have an index, citations, plots, math (for the theoretical background of machine learning models) and so on. I am going to use Python for the developing. However, I am still thinking about which sofwtare should I use for typing purposes. Regarding the first option, the main inconvenient is that coding has to be developed outside Latex (for testing) and then it has to be pasted in Latex, which results in a loss of hyphenation (at least that happened when I first used it) and the consequent loss of time for making it again.