text_plain_output_1.png
import matplotlib.pyplot as plt

import pandas as pd

train_full_set = pd.read_csv('../input/train.csv')

full_set_initial = pd.get_dummies(data=train_full_set, columns=['Embarked', 'Sex', 'Survived'], drop_first=True)
full_set_initial = full_set_initial.drop(['PassengerId', 'Name', 'Ticket', 'Cabin'], axis=1)
corr = full_set_initial.corr()
plt.colorbar()
plt.xticks(range(len(corr)), corr.columns, rotation='vertical')
plt.yticks(range(len(corr)), corr.columns)

test_full_set = pd.read_csv('../input/test.csv')

full_set = pd.concat([train_full_set, test_full_set])
full_set = full_set.reset_index(drop=True)

print(full_set.head(5))