pandas.DataFrame.plot.scatter

DataFrame.plot.scatter(x, y, s=None, c=None, **kwargs)[source]

Создайте диаграмму рассеяния с различным размером и цветом маркеров.

Координаты каждой точки определяются двумя столбцами DataFrame, и для представления каждой точки используются заполненные кружки. Такой вид графика полезен для выявления сложных корреляций между двумя переменными. Точки могут, например, представлять собой естественные 2D-координаты, такие как долгота и широта на карте, или, в общем случае, любые пары метрик, которые можно построить друг против друга.

Параметры:

x:int или str

Имя столбца или позиция столбца, используемые в качестве горизонтальных координат каждой точки.

y:int или str

Имя столбца или позиция столбца, используемые в качестве вертикальных координат каждой точки.

s:str, скаляр или массив-подобный объект, необязательно

Размер каждой точки. Возможные значения:

• Строка с именем столбца, используемого для размера маркера.

• Один скаляр, чтобы все точки имели одинаковый размер.

• Последовательность скаляров, которая будет использоваться для размера каждой точки рекурсивно. Например, при передаче [2,14] размер всех точек будет либо 2, либо 14, поочередно.

c:str, int или массив-подобный объект, необязательно

Цвет каждой точки. Возможные значения:

• Один строковый цвет, указанный по имени, в RGB или RGBA формате, например «красный» или «#a98d19».

• Последовательность строковых цветов, указанных по имени, в RGB или RGBA формате, которая будет использоваться для цвета каждой точки рекурсивно. Например, [‘зеленый’,’желтый’] все точки будут заполнены зеленым или желтым цветом, поочередно.

• Имя столбца или позиция, значения которых будут использоваться для окрашивания точек маркеров в соответствии с цветовой картой.

**kwargs

Ключевые аргументы для передачи в DataFrame.plot().

Возвращает:

matplotlib.axes.Axes или массив numpy из них

См. также

matplotlib.pyplot.scatter

Диаграмма рассеяния с использованием различных форматов входных данных.

Примеры

Посмотрим, как построить диаграмму рассеяния, используя координаты из значений в столбцах DataFrame.

>>> df = pd.DataFrame([[5.1, 3.5, 0], [4.9, 3.0, 0], [7.0, 3.2, 1],
...                    [6.4, 3.2, 1], [5.9, 3.0, 2]],
...                   columns=['length', 'width', 'species'])
>>> ax1 = df.plot.scatter(x='length',
...                       y='width',
...                       c='DarkBlue')

И теперь с цветом, определяемым также столбцом.

>>> ax2 = df.plot.scatter(x='length',
...                       y='width',
...                       c='species',
...                       colormap='viridis')