Question

我想拥有一个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数组呢？

感谢。

Answer 1

您需要使用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)

拥有一个dtype=object的numpy数组可能会导致误导，因为使用标准dtypes进行操作的强大的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

Answer 2

我相信这是两个库之一的错误。我也相信这是固定的。

输入：

import sys
import numpy as np
import gmpy2

print(sys.version)
print(np.__version__)
print(gmpy2.version)

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

输出：

3.4.2 (v3.4.2:ab2c023a9432, Oct  6 2014, 22:15:05) [MSC v.1600 32 bit (Intel)]
1.9.1
2.0.5
[[mpfr('1.0') mpfr('1.0') mpfr('1.0') mpfr('1.0') mpfr('1.0')]
 [mpfr('1.0') mpfr('1.0') mpfr('1.0') mpfr('1.0') mpfr('1.0')]
 [mpfr('1.0') mpfr('1.0') mpfr('1.0') mpfr('1.0') mpfr('1.0')]
 [mpfr('1.0') mpfr('1.0') mpfr('1.0') mpfr('1.0') mpfr('1.0')]
 [mpfr('1.0') mpfr('1.0') mpfr('1.0') mpfr('1.0') mpfr('1.0')]]

Numpy在遇到未知类型时没有正确说明该做什么，或者gmpy2没有指定如何按某种方式划分（__rdiv__）。

除非您打算在其元素上进行书写，否则无需指定dtype的{{1}}。乘法等操作会产生新的ndarray，而Numpy会计算出ndarray要使用的内容。

具有mpz / mpfr值的numpy数组

2 个答案: