我编写了一个自定义类IntegerMod,用于对以质数为模的整数进行编码.一切正常.然后将它们用作使用numpy.poly1d构建的多项式的系数,并设法在IntegerMod中实现足够的方法,以便使用这些多项式进行所需的运算(例如,找到一堆给定的插值多项??式).
剩下的只是一点点细节,那就是这些多项式的print(pol)实际上失败了,因为Python尝试使用字符串格式%g来表示系数,并且未能说IntegerMod应该是字符串或数字.
实际上,IntegerMod继承自number.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'
"""