当使用自定义类实例作为索引时，我不明白 numpy ndarray 的索引是如何工作的。
我有以下代码：
import numpy as np
class MyClass:
def __index__(self):
return 1,2
foo = np.array([[1,2,3],[4,5,6]])
bar = MyClass()
print(foo[1,2])
print(foo[bar])
我希望从两个打印函数中得到相同的结果 (6)。但是从第二个使用类实例的索引开始，我收到一个错误：
IndexError: only integers, slices (`:`), ellipsis (`...`), numpy.newaxis (`None`) and integer or boolean arrays are valid indices
如果我显式调用 __index__ 方法
print(foo[bar.__index__()])
有用。但这违背了魔法方法的目的。
如果我只使用一个索引调用数组，则一切正常：
import numpy as np
class MyClass:
def __index__(self):
return 1
foo = np.array([[1,2,3],[4,5,6]])
bar = MyClass()
print(foo[1])
print(foo[bar])
>>> [4 5 6]
>>> [4 5 6]
所以我没有得到：
我错过了什么，还是 ndarray 不支持这种索引？
我只想补充一点， __index__ 方法如何输出其结果显然无关紧要。我试过了：
return a, b
return (a, b)
return tuple((a, b))
他们都没有为我工作。
正如这里提到的，__index__方法Must return an integer.
这就是为什么您的尝试无效，而“一个索引”示例有效的原因。
本文收集自互联网，转载请注明来源。
如有侵权，请联系[email protected] 删除。
我来说两句