我为一个素数的整数编写了一个自定义类IntegerMod。一切正常。然后将它们用作用numpy.poly1d构建的多项式的系数,并设法在IntegerMod中实现足够的方法,以便对这些多项式的运算按我的需要进行(例如,找到给定一组点的插值多项式)。
只剩下一点细节,那就是那些多项式的print(pol)实际上失败了,因为Python试图对coeficients使用字符串格式化%g,而这并不能说明IntegerMod应该是一个字符串或一个数字。
实际上IntegerMod继承自numbers.Number,但似乎还不够我的问题是,我能在我的类中实现字符串格式化的行为吗?如果没有,我该如何处理这个问题?
产生误差的mwe:

import numbers
import numpy as np


class IntegerMod(numbers.Number):
    def __init__(self, k, p):
        self.k = k % p
        self.p = p

    def __repr__(self):
        return "<%d (%d)>" % (self.k, self.p)

if __name__ == "__main__":
    p = 13
    coef1 = IntegerMod(2, p)
    coef2 = IntegerMod(4, p)
    print(coef1)  # Works as expected
    pol = np.poly1d([coef1, coef2])
    print(pol)
    """ # error:
        s = '%.4g' % q
        TypeError: float() argument must be a string or a number, not 'IntegerMod'
    """

最佳答案

也许您应该实现__float__方法,因为poly1d格式需要浮点。
像这样的东西

    def __float__(self):
        return float(self.k)

10-01 15:45