Ask Your Question

Pulling the index of an entry of a matrix

asked 2013-05-21 12:18:11 +0200

JoshIzzard gravatar image

Let $M$ be some matrix, then I am looking for how to find out which entries of $M$ make the statement M[i,j] == 25 true. Specifically, this is the code I'm using to generate the matrix, called deg:

deg = matrix(ZZ, 15)
A = lambda i: WeylCharacterRing("A{0}".format(i), style = "coroots")
for i in range(1,15):
    fw = A(i).fundamental_weights()
    for j in range(1,len(fw)):
        deg[i,j] = A(i)(fw[j]).degree()

print deg

It works great, but now instead of printing deg and finding the entries manually, I would like to be able to store which entries [i,j] of deg are equal to 25 in some list, and then print that list, which would ideally look something like:

{ (2,3), (4,4) }


Any tips on where to look or help files to examine?

Thanks very much.

edit retag flag offensive close merge delete

2 Answers

Sort by ยป oldest newest most voted

answered 2013-05-21 13:44:30 +0200

niles gravatar image

One way you could do this is to use a conditional list comprehension; these take the form

[x for x in list if <condition>]

And since you will be iterating over a matrix, a double list comprehension is in order; these take the form

[x for b in a for x in b]

where a is the "outer list", and b is the "inner list". And since you want to know the indices and the entries together, we'll use the enumerate function. So here's a conditional double list comprehension which returns (i,j) such that deg(i,j) == 15:

sage: [(i,j) for (i,deg_i) in enumerate(deg) for (j,deg_ij) in enumerate(deg_i) if deg_ij == 15]
[(5, 2), (5, 4), (14, 1)]

Another Sage-specific data type that might interest you is sets -- these are like lists, but automatically condense duplicate entries. For example, the set of values in the matrix deg can be given by

sage: deg_values = set([x for row in deg for x in row])
sage: len(deg_values)
edit flag offensive delete link more


Thanks very much for the help @niles, this should work well.

JoshIzzard gravatar imageJoshIzzard ( 2013-05-21 15:24:34 +0200 )edit

answered 2023-11-05 11:53:11 +0200

Sylvain gravatar image

updated 2023-11-05 11:53:48 +0200

Matrix type has a find method which allows that:

sage: deg.find(lambda entry:entry==15,indices=True)
{(5, 2): 15, (5, 4): 15, (14, 1): 15}

It returns a dictionary of the indices and the values of the matrix matching the condition given as input. The condition is a function.

edit flag offensive delete link more


Nice bit of necromancy (the question is 10 years (!) old)... Thank you on behalf of future (per-)users of

Emmanuel Charpentier gravatar imageEmmanuel Charpentier ( 2023-11-05 13:47:24 +0200 )edit

Your Answer

Please start posting anonymously - your entry will be published after you log in or create a new account.

Add Answer

Question Tools

1 follower


Asked: 2013-05-21 12:18:11 +0200

Seen: 597 times

Last updated: Nov 05 '23