Я только что узнал, почему это происходит. Оперативная память colab использовалась полностью при прохождении через последнюю ячейку. Это случилось, когда я использовал график matplotlib, но когда я использовал гистограммы matplotlib, все получилось просто отлично. Гистограммы, вероятно, легче, если вы используете значения с плавающей точкой высокой точности. Я повторил несколько испытаний между двумя методами, и каждый раз получал один и тот же результат.