изменение функции построения графиков scipy stats.probplot с помощью matplotlib

Question

Я не эксперт по matplotlib, поэтому мне трудно пытаться установить параметры статистики scipy.

Мой код берет столбец pandas df, перебирает столбцы и пытается построить значения столбцов с помощью функции stats.probplot. Это мой код:

plt.figure(figsize=(10,5))
for col in model_predictions.columns:
    res = stats.probplot(df[col]), plot=plt)
    plt.legend = col
plt.show()

Это генерирует диаграммы, которые я хочу, но трудно читать (без легенд, одинаковых цветов). Помимо нанесения их друг на друга, я хотел бы нарисовать каждую линию другим цветом, а также добавить легенду для каждой линии, равную str в col. Есть ли способ сделать это?

Я всегда могу взять выходные данные функции кортежа, запустить ее с другим новым определением и добавить выходные данные к новому pandas df (для последующего построения с большим контролем); но мне было интересно, есть ли более быстрый путь.

Спасибо

Answer 1

Вы можете построить их вручную, взяв вывод stats.probplot, то есть:

from scipy.stats import probplot

for col in model_predictions.columns
    plt.plot(*stats.probplot(df[col])[0], label=col)
plt.legend(loc='best')
plt.show()