Возможно, вы могли бы использовать ggplot2 в Python с пакетом plotnine вместе с функцией geom_point().
Вот код в R, чтобы продемонстрировать, как он работает , Процедуры должны быть идентичны в Python (хотя и с вышеуказанными изменениями в грамматике от R до Python):
> library(ggplot2)
> p <- ggplot(mtcars, aes(x=wt, y=mpg, color=cyl, size=cyl))
> p <- p + geom_point(shape=15, fill=NA)
> print(p)
Здесь фрейм данных mtcars (встроенный в R) имеет обычные x и y параметры. color точки задается непрерывным диапазоном значений в столбце cyl. size точки (радиус в случае круглой точки, но диагональ в случае квадрата) также определяется категориальными значениями в cyl.
В plotnine код может выглядеть следующим образом:
#!/usr/bin/env python
from plotnine import *
from plotnine.data import mtcars
p = (ggplot(mtcars, aes(x='wt', y='mpg', color='cyl', size='cyl')) +
geom_point(shape='s', fill=None))
p.save("../results/test.png")
Это дает следующий результат:
Сезон по вкусу: вместо использования фрейма данных mtcars вы будете использовать фрейм данных, который вы тянете с помощью pandas, наряду с настройкой необходимых столбцов для указания положения, цвета и размера точечных элементов.