python - 扩展 numpy 数组

Question

我是 Python/numpy 的新手。

我正在尝试扩展 numpy.array 以赋予它一些功能，使其能够更好地表示图像(例如转换为灰度等)。

import numpy as np
import cv2
from support import *
import matplotlib.pyplot as plt

class Frame(np.array):
    def __init__(self):
        print "new frame"


f = Frame()

目前这给了我:

  File "o.py", line 6, in <module>
    class Frame(np.array):
TypeError: Error when calling the metaclass bases
    cannot create 'builtin_function_or_method' instances

我不明白为什么这是 Python 的问题？

Answer 1

您想子类化 np.ndarray，而不是 np.array，但这比在您的示例中换成另一个要复杂一些。可能值得看一下文档: http://docs.scipy.org/doc/numpy/user/basics.subclassing.html