from IKZ import calc from numpy import arange a = int(input('Enter a value: ')) b = int(input('Enter b value: ')) res = [] xRange = arange(a - 3 * b**0.5, a + 3 * b ** 0.5, 0.2 * b ** 0.5) for i in range(len(xRange)): res.append(calc(xRange[i], a, b)) with open('IKZ_res.txt', 'w') as file: resString = '' for i in range(len(res)): divider = ',' if i % 2 == 0 else '\n' resString += f'{res[i]}{divider}' file.write(resString)