text_plain_output_1.png
"""def correlation(x_data, threshold):    
    col_corr = set()
    corr_matrix = x_data.corr()
    for i in range(len(corr_matrix.columns)):
        for j in range(i):
            if abs(corr_matrix.iloc[i, j] > threshold:
                    colname = corr_matrix.columns[i]   
                    col_corr.add(colname)
return col_corr"""