Python generate a mask for the lower triangle of a matrix

Question

I have this code from seaborn documentation to generate a mask for the upper triangle of a given correlation matrix

# Compute the correlation matrix
corr = d.corr()

# Generate a mask for the upper triangle

mask = np.zeros_like(corr, dtype=np.bool)
mask[np.triu_indices_from(mask)] = True

how would one achieve the invert, a mask for the lower triangle?

score 11 · Accepted Answer · answered Dec 17 '18 at 20:04

11

Simply replace triu_indices_from with tril_indices_from:

mask = np.zeros_like(corr, dtype=np.bool)
mask[np.tril_indices_from(mask)] = True

answered Dec 17 '18 at 20:04

MEE

2,114
17
21

yatu · Answer 2 · 2018-12-17T20:08:35.100

Take the transpose of your matrix:

mask = mask.T

mask
array([[ True, False, False, False, False],
       [ True,  True, False, False, False],
       [ True,  True,  True, False, False],
       [ True,  True,  True,  True, False],
       [ True,  True,  True,  True,  True]])

mask.T
array([[ True, False, False, False, False],
       [ True,  True, False, False, False],
       [ True,  True,  True, False, False],
       [ True,  True,  True,  True, False],
       [ True,  True,  True,  True,  True]])

However this is more of a workaround, the correct solution is @john 's

score 1 · Answer 3 · answered Dec 17 '18 at 20:03

1

You can simply transpose the mask that you have:

mask = np.zeros_like(corr, dtype=np.bool).T
mask[np.triu_indices_from(mask)] = True

answered Dec 17 '18 at 20:03

Tim

2,756
1
15
31

Python generate a mask for the lower triangle of a matrix

3 Answers3

Linked