我为一个素数的整数编写了一个自定义类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)