Python - 如何实现与 NumPy 函数兼容的自定义类？

Question

我正在尝试实现我自己的自定义类来存储数据。我想让它与 NumPy 兼容，这样我就可以像这样调用 NumPy 函数:

np.sin(my_object)

我知道有一本名为__array_interface__的字典，但是在尝试使用它时我遇到了很多奇怪的错误。

import numpy as np
import pandas as pd

class TDF:
    __array_interface__ = {'typestr': '|i1', 'version': 1}

    def __init__(self):
        self.ddata = pd.DataFrame([1, 2, 3])
        self.shape = self.ddata.shape

    def __iter__(self):
        return iter(self.ddata)

    def __len__(self):
        return len(self.ddata)

    def __getitem__(self, key):
        return self.ddata.__getitem__(key)

if __name__ == '__main__':
    tdf1 = TDF()
    tdf = np.sin(tdf1)

上面的代码给了我一个运行时错误:

ValueError: setting an array element with a sequence.

我错过了什么？另一方面，pandas 的源代码(哪些类与 NumPy 兼容)没有显式使用 array_interface 字典...

Answer 1

一个简单的修复方法是实现

def __array__(self):
    return self.ddata