numpy의 하위 클래스 배열

Question

문제가 있습니다. 문제는 : numpy 배열의 하위 클래스를 만들고 해당 유형의 객체 배열을 만들고 싶습니다. 해당 배열의 항목을 참조 할 때 여전히 그 하위 클래스의 인스턴스가되고 싶습니다. 대신, 그것은 numpy 배열의 인스턴스입니다.

import numpy as np 


class ImageWrapper(np.ndarray): 

    def __new__(cls, image_data): 
     assert image_data.ndim in (2, 3) 
     return image_data.view(cls) 

    @property 
    def n_colours(self): 
     return 1 if self.ndim==2 else self.shape[2] 


n_frames = 10 
frames = [ImageWrapper(np.random.randint(255, size = (20, 15, 3)).astype('uint8')) for _ in xrange(n_frames)] 
video = np.array(frames) 
assert video[0].n_colours == 3 

나에게 제공합니다 : AttributeError를 'numpy.ndarray'개체가 어떤 속성 'n_colours'가

가 어떻게이 작품을 만들 수 있습니다

가 여기에

실패 테스트입니다?

상황이 이미 시도 :

• 설정 subok = 비디오를 구성 할 때 진정한 -이 단지 작품 서브 클래스 객체의 단일 인스턴스가 아닌 목록에서 배열을 구성.
• 설정 DTYPE = 개체 또는 DTYPE = ImageWrapper 내가 그냥 비디오 목록을 만들 수 있음을 인식

작동하지 않지만, 다른 이유로 NumPy와 배열로 유지하는 것이 바람직 할 것이다.

Answer 1

numpy.array는이 사건을 처리 할 정도로 정교하지 않습니다. subok=True은 하위 클래스를 전달하는 함수를 알려주지 만 ndarray의 하위 클래스를 전달하지 않으면 목록을 전달합니다 (ndarray 하위 클래스의 인스턴스로 채워짐).

import numpy as np 


class ImageWrapper(np.ndarray): 

    def __new__(cls, image_data): 
     assert 2 <= image_data.ndim <= 4 
     return image_data.view(cls) 

    @property 
    def n_colours(self): 
     return 1 if self.ndim==2 else self.shape[-1] 


n_frames = 10 
frame_shape = (20, 15, 3) 
video = ImageWrapper(np.empty((n_frames,) + frame_shape, dtype='uint8')) 
for i in xrange(n_frames): 
    video[i] = np.random.randint(255, size=(20, 15, 3)) 
assert video[0].n_colours == 3 

주의 사항 4 차원 배열을 입력으로 허용하려면 ImageWrapper를 업데이트해야합니다.

Answer 2

달성하고자하는 것이 무엇이든, 아마도 ndarray를 서브 클래 싱하는 것보다 더 나은 방법이있을 것입니다. 그러나 배열을 만들 때 조심해야하지만 배열의 유형이 object 일 수 있습니다.

>>> video = np.empty((len(frames),), dtype=object) 
>>> video[:] = frames 
>>> video[0].n_colours 
3 

을하지만이되지 않습니다 :이 작동

>>> video = np.array(frames, dtype=object) 
>>> video[0].n_colours 
Traceback (most recent call last): 
    File "<stdin>", line 1, in <module> 
AttributeError: 'numpy.ndarray' object has no attribute 'n_colours'