如何提取mpmath返回结果中的数值以用于后续计算？

我没有使用mpmath包的经验，最近用它做一些小测试时得到了以下返回结果：

from mpmath import *
loggamma(3+2j)
mpc(real='-0.031639059373961193', imag='2.022193197501327')
atan2(2,3)
mpf('0.5880026035475675')

我想了解如何提取mpmath返回结果中的数值（-0.031…、2.022…、0.588…），以便用于后续计算。

补充说明 - 例如使用numpy时，我能直接得到可直接使用的实数：

import numpy as np
np.arctan2(2,3)
0.5880026035475675

那么，使用mpmath实现同样效果的最佳方法是什么？

回答 #

mpmath返回的mpf（任意精度浮点数）和mpc（任意精度复数）对象本身就支持直接参与后续计算，不需要额外转换。如果确实需要提取成Python原生的数值类型，可以用以下方法：

1. 处理mpf类型结果 #

直接用float()转换为Python原生浮点数，或者直接用mpf对象参与运算：

from mpmath import *
result = atan2(2,3)
# 直接参与高精度计算
print(result * 2)  # 输出 mpf('1.176005207095135')
# 转换为原生float
native_float = float(result)
print(native_float)  # 输出 0.5880026035475675

2. 处理mpc类型结果 #

mpc对象自带real和imag属性，可直接访问实部和虚部；也可以用complex()转换为原生复数类型：

from mpmath import *
complex_result = loggamma(3+2j)
# 直接获取实部、虚部并计算
real_part = complex_result.real
imag_part = complex_result.imag
print(real_part + imag_part)  # 输出 mpf('1.9905541381273658')
# 转换为原生complex
native_complex = complex(complex_result)
print(native_complex)  # 输出 (-0.031639059373961193+2.022193197501327j)

注意事项 #

• 如果需要保留mpmath的高精度特性，建议直接使用mpf和mpc对象参与运算，转换为原生float/complex会丢失精度。
• 大部分Python原生运算操作都能直接作用于mpf和mpc对象，使用逻辑和原生数值类型一致。

内容的提问来源于stack exchange，提问作者Andrew