if len(sys.argv) > 2: name = sys.argv[2] else: name = 'prime_numbers_1.csv' else: print("Using default value N_max = 100") print("Optional command line argument to change this") N_max = 100 name = 'prime_numbers_1.csv' # setup other default parameters var_names = ['i', 'pn'] path = os.getcwd() + '/' logger = DataLogger(path, name, var_names, log_interval=99) def is_prime(n): for ni in range(2, n): if (n % ni) == 0: return False return True index = 0 for i in range(N_max): if is_prime(i): print(index, i) logger.log(i=index, pn=i) index += 1 logger.flush_cache()