Как создать матрицу рассеяния в pandas (с примерами)

Матрица диаграммы рассеяния — это именно то, на что она похожа: матрица диаграммы рассеяния.

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

Вы можете использовать функцию scatter_matrix() для создания матрицы рассеяния из DataFrame pandas:

 p.d. plotting . scatter_matrix (df)

В следующих примерах показано, как использовать этот синтаксис на практике со следующим DataFrame pandas:

 import pandas as pd
import numpy as np

#make this example reproducible
n.p. random . seeds (0)

#createDataFrame
df = pd. DataFrame ({' points ': np.random.randn(1000),
                   ' assists ': np.random.randn(1000),
                   ' rebounds ': np.random.randn(1000)})

#view first five rows of DataFrame
df. head ()

	points assists rebounds
0 1.764052 0.555963 -1.532921
1 0.400157 0.892474 -1.711970
2 0.978738 -0.422315 0.046135
3 2.240893 0.104714 -0.958374
4 1.867558 0.228053 -0.080812

Пример 1: Базовая матрица диффузии

Следующий код показывает, как создать базовую матрицу дисперсии:

 p.d. plotting . scatter_matrix (df)

Пример 2: Матрица рассеяния для определенных столбцов

Следующий код показывает, как создать матрицу рассеяния только для первых двух столбцов DataFrame:

 p.d. plotting . scatter_matrix (df. iloc [:, 0:2])

Пример 3: Матрица рассеяния с пользовательскими цветами и интервалами

Следующий код показывает, как создать матрицу рассеяния с настраиваемыми цветами и определенным количеством групп для гистограмм:

 p.d. plotting . scatter_matrix (df, color=' red ', hist_kwds={' bins ': 30 , 'color': ' red '})

Пример 4: Матрица дисперсии с графиком KDE

Следующий код показывает, как создать матрицу дисперсии с графиком оценки плотности ядра вдоль диагоналей матрицы вместо гистограммы:

 p.d. plotting . scatter_matrix (df, diagonal=' kde ')

Полную онлайн-документацию функции scatter_matrix() вы можете найти здесь .

Дополнительные ресурсы

В следующих руководствах объясняется, как создавать другие распространенные диаграммы в Python:

Как создать тепловые карты в Python
Как создать колоколообразную кривую в Python
Как создать оживальную диаграмму в Python