Not sure how you can test the functions to make sure it covers different cases and put the functions in the notebook into production. But if what you want is to play and explore the data, by no means, keep using the notebooks. I also created small functions in my notebook when I just want to explore the data.