`[i for i in range(n_features)]` is `list(range(features))`. But `np.array([i for i in range(n_features)])` is `np.arange(features)`.