немного подправить код

есть код на пайтон. он делает расчет для одного зерна, т.е. записываются данные и строится график.

нужно сделать так чтобы он делал расчет для 5 зерен, т.е. данные записываются в 5 разных файлов и на их основе уже строится график с 5 прямыми