Getting attributes from arrays of objects in NumPy

Question

Let\'s say I have an class called Star which has an attribute color. I can get color with star.color.

But what if I have a Num

Answer 1

You could use numpy.fromiter(s.color for s in stars) (note lack of square brackets). That will avoid creating the intermediate list, which I imagine you might care about if you are using numpy.

(Thanks to @SvenMarnach and @DSM for their corrections below).