File "C:/Users/Benji/PycharmProjects/Code/ANOTHER22.py", line 57, in generator
x_class[np.array(y)[i]].append(np.array(x)[i])
TypeError: only integer scalar arrays can be converted to a scalar index
x_class = [[] for i in range(len(np.unique(y)))]
for i in range(len(x)):
x_class[y[i]].append(x[i])