sim_mat2.py

sim_mat2.py

def upper(df):
    '''Returns the upper triangle of a correlation matrix.

    You can use scipy.spatial.distance.squareform to recreate matrix from upper triangle.

    Args:
      df: pandas or numpy correlation matrix

    Returns:
      list of values from upper triangle
    '''
    try:
        assert(type(df)==np.ndarray)
    except:
        if type(df)==pd.DataFrame:
            df = df.values
        else:
            raise TypeError('Must be np.ndarray or pd.DataFrame')
    mask = np.triu_indices(df.shape[0], k=1)
    return df[mask]

# Compare that the upper triangle is properly extracted. 
print("Extracted upper triangle from m1")
print(np.round(upper(m1)[:25],3))
print("Top 3 rows from m1")
display(m1.iloc[:3, :].round(3))