Я пытаюсь построить массив, созданный одним горячим кодированием для нуклеотидных последовательностей. последовательности были положительными и отрицательными, и я хочу изменить цвет диаграммы рассеяния двумя разными цветами. Не уверен, что здесь не так
dim1 = pca_result[:,0]
dim2 = pca_result[:,1]
plt.figure(figsize=(10,10))
plt.scatter(dim1, dim2, cmap= 'YlGnBu' ,c=search_reg['NR+/-'], s=1)
plt.title("PCA for Negative data")
plt.xlabel('PCA 1')
plt.ylabel('PCA 2')
plt.show()
dim1 и dim 2
являются массивами, в то время как search_reg['NR+/-'] содержит значения 0 и 1.
График на графике черно-белый, но из-за белого фона он плохо виден.
Как я могу изменить цвет графиков?