up vote 0 down vote favorite
share [g+] share [fb]

I have a multi-dimensional array of objects. I want to interate over the objects using the nditer iterator. Here is a code example:

import numpy as np

class Test:
    def __init__(self,a):
        self.a = a
    def get_a(self):
        return self.a

b = np.empty((2,3),dtype = object)

t_00 = Test(0)
t_01 = Test(1)
t_11 = Test (11)

b[0,0] = t_00
b[0,1] = t_01
b[1,1] = t_11

for item in np.nditer(b,flags = ["refs_ok"]):
    if item:
        print item.get_a()

I would expect the "item" to contain the object reference that I can use to access data. However I am getting the following error:AttributeError: 'numpy.ndarray' object has no attribute 'get_a' My question is how can I go thru the array to access the object in the array?

link|improve this question
46% accept rate
One issue I see in the code is that get_a needs to return self.a, not a. – ditkin Jan 6 at 9:24
You are right. However, I have the basic problem that I cannot access the "test object" at all. It seems that each "item" is referring to a numpy.ndarray – user963386 Jan 6 at 9:42
feedback

1 Answer

up vote 0 down vote accepted

The array.flat method of iteration will work, and can confirm that this works as you'd expect

for item in b.flat:
    if item:
        print item.get_a()
link|improve this answer
Thanks Jason! However I am still trying to find out how to use the nditer iterator. It works all right if I have an array with regular numbers. When I have objects, there is a problem. But I can see that nditer is reurning the right object address but may be with the wrong type. – user963386 Jan 6 at 22:11
feedback

Your Answer

 
or
required, but never shown
discard

Not the answer you're looking for? Browse other questions tagged or ask your own question.