Корреляция между тремя переменными в диаграмме рассеивания

Question

Мне нужно получить график корреляции между Возрастом, классом билета и выжившими людьми, но я не понимаю как это сделать, так как там три переменные Код который у меня сейчас:

fig, ax = plt.subplots(figsize=(8, 6))
ax.scatter(x = dataset['Pclass'], y = dataset['Age'])
plt.xlabel("Pclass", size=15)
plt.ylabel("Age", size=15)
plt.title('Pclass and Age vs Survived', size=18)
ax.grid(axis = 'y')
plt.xticks([1,2,3])
plt.show()

Answer 1

Можно использовать Seaborn:

import numpy as np
import pandas as pd
import seaborn as sns

n = 200
df = pd.DataFrame({'Pclass': np.random.randint(1, 4, n), 
                   'Age': np.random.randint(10, 100, n), 
                   'Survived': np.random.randint(0, 2, n)});
sns.swarmplot(x='Pclass', y='Age', hue='Survived', data=df, dodge=True);

БЛОГ НА HUSL

Корреляция между тремя переменными в диаграмме рассеивания

Ответы (1 шт):