np.ndarrayのクラス継承で値を保持したnp.ndarrayを使いたい

Question

前提・実現したいこと

np.ndarrayを継承してコンテナ型の挙動を変更したい

発生している問題・エラーメッセージ

クラスを継承しても値を代入できない。代入方法がわからない。

該当のソースコード

import numpy as np


class Board(np.ndarray):
	def __init__(self, board):
		super().__init__()


Board(np.array([0,1,2]))

試したこと

selfを変更してみる

import numpy as np


class Board(np.ndarray):
	def __init__(self, board):
		super().__init__()
		self = board


board = Board(np.array([0,1,2]))
print(board)
#>>>[]

補足情報（FW/ツールのバージョンなど）

Python 3.9.2
Numpy 1.21.3

Answer 1

Slightly more realistic example - attribute added to existing array を参考にするとよろしいかと思います。

import numpy as np

class Board(np.ndarray):
    def __new__(cls, input_array):
        obj = np.asarray(input_array).view(cls)
        return obj

board = Board([0, 1, 2])
print(board)
print(type(board))
print(Board.__bases__)

#
[0, 1, 2]
<class '__main__.Board'>
(<class 'numpy.ndarray'>,)

Comments

[sakuramochi_py]

回答ありがとうございます！