Пожалуйста, смотрите код ниже. Я пытаюсь получить 2 одинаковых графика рассеяния, в ax2 я пытаюсь установить цвет после создания графика рассеяния. Как я могу добиться этого?
Я хочу этого, потому что в моем интерфейсе я пытаюсь, чтобы пользователь (опционально) выбирал данные для окрашивания диаграммы рассеяния. Я мог бы просто переделать весь график, но я предполагаю, что для большого количества точек данных лучше добавить цвета к существующему объекту оси. Это правильно?
import numpy as np
import matplotlib.pyplot as plt
fig, (ax1,ax2) = plt.subplots(1,2)
x = np.random.rand(100)
y = np.random.rand(100)
z = np.random.rand(100)
#in ax1, I set color using the 'c' argument
ax1.scatter(x,y, c=z)
sc = ax2.scatter(x,y)
#in ax2, I try to mimic the 'c' argument with set_color but it raises error
# sc.set_color(z)