import pickle

import numpy as np

with open("Y.dat", "rb") as file:
    Y = pickle.load(file)
N = Y.shape[1]
X = np.arange(N)
with open("X.dat", "wb") as file:
    pickle.dump(X, file)