fast way to get the indices of a lower triangular matrix as 1 dimensional list in python

Question

Given the number of rows (or columns) , n, of a square matrix, I am trying to get the index pairs of the lower triangular matrix in a 1 dimensional list. So far I thought of the following solution:

def getLowerTriangularIndices(n):
    inds=[];         
    for i in range(1,n):
        for j in range(i):
            inds.append((i,j))
    return inds;

Considering the two for loops, it would be far better to have a more efficient way of calculating this maybe using numpy. Does anyone have a suggestion?

score 7 · Accepted Answer · edited Oct 11 '13 at 12:23

7

Numpy has a method for that...

import numpy as np

# create your matrix. If it's not yet a numpy array, make it one
ar = np.array(matrix)
indices = np.tril_indices_from(ar)

This returns a tuple of two arrays. If you want to have them as lists, you could do

indices = [list(x) for x in np.tril_indices_from(ar)]

You actually do not need to have an array to get the indices, there is also np.tril_indices, which takes the shape as arguments.

So your function would read:

def getLowerTriangularIndices(n):
    return [list(x) for x in np.tril_indices(n)]

or if you want a list of tuples instead:

def getLowerTriangularIndices(n):
    return zip(*np.tril_indices(n)]

edited Oct 11 '13 at 12:23

BioGeek

21,897
23
83
145

answered Jan 11 '13 at 08:49

Thorsten Kranz

12,492
2
39
56

1

You actually don't even need the array to do this. `np.tril_indices` takes an integer for the dimension of a matrix and returns the tuple of indices directly – talonmies Jan 11 '13 at 08:53
I was editing the exact same thing while you commented - so +1 for you. – Thorsten Kranz Jan 11 '13 at 08:56
1

I think you want just a single argument in tril_indices as tril_indices(n,n) will return the whole array. – Chris Ferrie Oct 08 '13 at 22:50
You're right - interesting that nobody discovered this mistake earlier. Fixed my answer. Thank you! – Thorsten Kranz Oct 10 '13 at 08:55

fast way to get the indices of a lower triangular matrix as 1 dimensional list in python

1 Answers1