是否可以通过 njit 函数中的名称更改类变量？

Question

我试图通过它的名字来修改一个类的变量，所以基本上我所做的就是调用setattr函数。我的问题是当我试图用numba njit装饰器做到这一点时，它不再起作用了。

那么有没有办法在里面做同样的事情numba？

import numba as nb

class A():

    def __init__(self):

        self.a =0.

@nb.njit()

def test(A,s):

    setattr(A,s,1)

A = A()

s = 'a'

print(A.a)

test(A,s)

print(A.a)

更新

是否可以test在@nb.njit装饰器之外的函数中调用函数。在下面的例子中，var_update(A,s,1)函数无法编译？

import numba as nb

class A():

    def __init__(self):

        self.a =0.

    def set_a(self,a):

        self.a=a

@nb.njit()

def test(A,s):

    var_update(A,s,1) # this function could not be compiled?

def var_update(Obj, s , val):

    setattr(Obj,s,val)

CA = A()

s = 'a'

print(CA.a)

test(CA,s)

print(CA.a)

Answer 1

是否可以通过 njit 函数中的名称更改类变量？不，目前（使用 numba 0.43.1）使用 nopython-mode numba 函数是不可能的。它在您的示例中不起作用的原因有两个：在 nopython 模式下，numba 只支持非常有限的类。numba.jitclass不支持自定义类（除了）。setattrnumba 根本不支持（请参阅numba 文档中“支持的 Python 功能”中的“内置函数”部分）。但是，如果您对如何设置自定义类的属性（但不是动态名称）感兴趣，您可以使用jitclass：import numba as [e-mail]([('a', nb.float64)])class A():    def __init__(self):        self.a = [e-mail] test(instance):    instance.a = 1A = A()print(A.a)test(A)print(A.a)请注意，如果您想使用自定义类，setattr那么我的建议是在 numba 函数之外执行此操作。Numba 非常适合数字运算和数组处理，但它不是通用工具！如果您想要一个更通用的工具，那么 Cython 可能会更好。以我的经验：如果它不是关于循环并且不涉及数字或数组，那么不要指望 numba 更有效 - 这当然是过度简化但在过去为我提供了很好的指导（例如参见我的回答“什么时候 numba 有效？”）。