Нужно сделать так, чтобы код был читаем, конкретно:
- оформление - мы не пишем код в jupyter notebook'ах, для этого лучше использовать pycharm, там можно удобно дебажить, а еще он умеет просто кучу всего. Для лучшего понимания оформления кода лучше ознакомиться с https://www.python.org/dev/peps/pep-0008/ (если хотите писать на питоне - очень рекомендуется к прочтению, будущие коллеги вас сразу полюбят), уверен, что есть где-то даже на русском
- написание функций: лучше делать функции "чистыми", т.е. так чтобы они не изменяли значения своих аргументов, а также использовали только их для вычисления своего результата - это выглядит ужасно, к тому же приводит к багам, которые очень трудно обнаружить. Если функция не должна ничего возвращать, не следует возвращать 0 - все функции по умолчанию возвращают None, этого достаточно.
- Аргументы функции: если у функции есть куча аргументов, то лучше сделать некоторые из них со значениями по-умолчанию - это избавит вызывающего от мучительного процесса придумывания