import pickle import numpy as np from k2c import k2c with open("Y.dat", "rb") as file: Y = pickle.load(file) M = Y.shape[0] K = (np.arange(M) + .5) / M with open("C.dat", "wb") as file: pickle.dump(k2c(K), file)