ученые тут пытаются эмулировать движение дыма/ччатсицы используя суперкомпьютеры для обсчета всего и вся, а ты взял и в 80 строк кода на питоне описал "физику чистой воды"
другими словами, ты должен был просчитать шанс попадания молнии в каждую точку пространства/плоскости/поверхности (с определенной дискретизацией), и нарисовать молнию там где этот шанс максимальный с учетом всех вычислений переменных