text_plain_output_1.png
import pandas as pd

import seaborn as sns

import pandas as pd
df = pd.read_csv('../input/spaceship-titanic/train.csv')
test_df = pd.read_csv('/kaggle/input/spaceship-titanic/test.csv')

useless_cols = ['PassengerId', 'Name', 'Cabin']
df.drop(columns=useless_cols, inplace=True)

import seaborn as sns
corr_matrix = df.corr()

high_corr = set()
for i in range(len(corr_matrix.columns)):
    for j in range(i):
        if abs(corr_matrix.iloc[i, j]) > 0.7:
            colname = corr_matrix.columns[i]
            high_corr.add(colname)
print(high_corr)