# Maps from indices from X to indices in dist matrix
dist_idx_map = np.zeros(X.shape[0], dtype=int)        dist_idx_map[row_missing_idx] = np.arange(row_missing_idx.shape[0])