具有 mpz/mpfr 值的 numpy 数组

我想要一个带有 mpz/mpfr 值的 numpy 数组。因为我的代码：

import numpy as np
import gmpy2
A=np.ones((5,5));
print A/gmpy2.mpfr(1);

生成：

RuntimeWarning: invalid value encountered in divide
  print A/gmpy2.mpfr(1);
[[1.0 1.0 1.0 1.0 1.0]
 [1.0 1.0 1.0 1.0 1.0]
 [1.0 1.0 1.0 1.0 1.0]
 [1.0 1.0 1.0 1.0 1.0]
 [1.0 1.0 1.0 1.0 1.0]]

据我所知，不可能将 gmpy mpfr 转换为 numpy float64。那么我怎样才能首先获得带有 mpfr 值的 numpy 数组呢？

Thanks.

您需要使用以下命令创建数组dtype=object，然后您可以在数组中使用任何 python 类型。我没有安装 gmpy2，但以下示例应该显示它是如何工作的：

In [3]: a = np.ones((5, 5), dtype=object)

In [5]: import fractions

In [6]: a *= fractions.Fraction(3, 4)

In [7]: a
Out[7]: 
array([[3/4, 3/4, 3/4, 3/4, 3/4],
       [3/4, 3/4, 3/4, 3/4, 3/4],
       [3/4, 3/4, 3/4, 3/4, 3/4],
       [3/4, 3/4, 3/4, 3/4, 3/4],
       [3/4, 3/4, 3/4, 3/4, 3/4]], dtype=object)

有一个 numpy 数组dtype=object可能会产生一点误导，因为强大的 numpy 机制使标准数据类型的操作速度超快，现在由默认对象的 python 运算符处理，这意味着速度将不再存在：

In [12]: b = np.ones((5, 5)) * 0.75

In [13]: %timeit np.sum(a)
1000 loops, best of 3: 1.25 ms per loop

In [14]: %timeit np.sum(b)
10000 loops, best of 3: 23.9 us per loop