K-means в dash plotly

Как отобразить нижепредставленный код для px.scatter?

X = df[['x','y']].to_numpy()
kmeans = KMeans(n_clusters=2).fit(X)
plt.scatter(X[:,0],X[:,1], c=kmeans.labels_, cmap='rainbow')

Если пишу,

px.scatter(X, color=kmeans.labels_)

то выводиться непрерывная расцветка. А мне нужно ее в дискертный вид привести.