[Numpy-discussion] Simplest ndarray subclass __new__ possible?
Colin J. Williams
cjw at sympatico.ca
Mon Feb 27 05:07:51 EST 2006
Zachary Pincus wrote:
> Hi folks,
>
> I'm trying to write an ndarray subclass with a constructor like the
> matrix constructor -- one which can take matrix objects, array
> objects, or things that can be turned into array objects.
>
> I've copied the __new__ method from matrix (and tried to eliminate
> the matrix-specific stuff), but there's a lot of code there. So I'm
> trying to figure out what the absolute minimum I need is for correct
> behavior. (This would be a useful wiki entry somewhere. In fact, a
> whole page about subclassing ndarray would be good.)
>
> What follows is what I have so far. Have I missed anything, or can
> anything else be removed?
>
> Zach
>
> class contour(numpy.ndarray):
> def __new__(subtype, data, dtype=None, copy=True):
>
> ##### Do I need this first if block?
> ##### Wouldn't the second block would do fine on its own?
> if isinstance(data, contour):
> dtype2 = data.dtype
> if (dtype is None):
> dtype = dtype2
> if (dtype2 == dtype) and (not copy):
> return data
> return data.astype(dtype)
>
> if isinstance(data, numpy.ndarray):
> if dtype is None:
> intype = data.dtype
> else:
> intype = numpy.dtype(dtype)
> new = data.view(contour)
> if intype != data.dtype:
> return new.astype(intype)
> if copy: return new.copy()
> else: return new
>
> # now convert data to an array
> arr = numpy.array(data, dtype=dtype, copy=copy)
>
> ##### Do I need this if block?
> if not (arr.flags.fortran or arr.flags.contiguous):
> arr = arr.copy()
>
> ##### Do I need the fortran flag?
> ret = numpy.ndarray.__new__(subtype, arr.shape, arr.dtype,
> buffer=arr, fortran=arr.flags.fortran)
> return ret
>
>
Would there be any merit in breaking this into two parts, __new__ to
allocate space and __init__ to initialize the data?
Colin W.
More information about the NumPy-Discussion
mailing list