Матрица с переменной длиной кусочка

Question

У меня есть матрица a с формой kx 3. И на основе первого индекса я хочу нарезать ее и сохранить.Первый индекс имеет значения 1: 1000.Хотя я закончил нарезку, я не знаю, как ее сохранить, так как длина срезов является переменной.Например, Y формы (163, 2), (55,2) и так далее.Как мне сохранить эти данные, чтобы я мог их вспомнить на основе таких индексов, как X [0], X [1], ..., X [999]?

def parseX(a, len):
  X = []
  for i in range(len):
    #Z = np.split(a, np.where(a[:, 0] == i+1))
    ### Split based on 0th column's value being (i+1)
    Y = a[ a[:,0] == i+1, :][:, [1,2]]
    print(Y.shape)
    #X[i] = Y
    #X[i].append(Y)
  #print(len(X))
  return X

Answer 1

Просто X.insert (i, Y) работает.

def parseX(a, len):
  X = []
  for i in range(len):
    ### Split based on 0th column's value being (i+1)
    Y = a[ a[:,0] == i+1, :][:, [1,2]]
    print(Y.shape)
    X.insert(i, Y)
  #print(len(X))
  return X