numpy.ndarray as Dictionary value

Question

I am storing numpy.ndarray as a value of a dictionary. My dictionary looks like:

{0: array([[ 1,  0, -1],
   [ 0, -1,  0],
   [ 1,  0,  0]])}

But when I am trying to find out if array exists in the dictionary using the following command

S = np.zeros((3,3), dtype=int)
if S in myDict.values():
   print "something"

I am getting the following error:

ValueError: The truth value of an array with more than one element is ambiguous. Use a.any() or a.all()

I have seen some one who have this problem, but their situation is different. Can anybody help me get ouf it?

At least any suggestion to check if a value exist in Dictionary? Thank you in advance.

What should be the command exactly?

user1763076
– user1763076

2014-05-04 01:54:06 +00:00
Commented May 4, 2014 at 1:54 — user1763076
– user1763076, Commented May 4, 2014 at 1:54

sshashank124 · Accepted Answer · 2014-05-04 01:54:43Z

1

You should use np.array_equal:

S = np.zeros((3,3), dtype=int)
if any(np.array_equal(S, i) for i in myDict.values()):
   print "Yes!"
else:
   print "Sorry. No."

[OUTPUT]
Sorry. No.

answered May 4, 2014 at 1:54

sshashank124

32.3k10 gold badges72 silver badges76 bronze badges

Sign up to request clarification or add additional context in comments.

2 Comments

sshashank124 Over a year ago

@vessilli, Glad I could help. If my answer was helpful, would you mind accepting it? (After the time limit) Thank you. And good luck.

user1763076 Over a year ago

Why not!! I will do it after the time limit.

Collectives™ on Stack Overflow

numpy.ndarray as Dictionary value

1 Answer 1

2 Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

2 Comments

Your Answer

Sign up or log in

Post as a guest

Related