python - 使用 python 的 long 类型的 numpy 数组

Question

我正在尝试创建一个类型为 long 的 numpy 数组(即 Python 的 long，而不是 numpy 的 long = int64)。

如果我这样做:

m = np.eye(6, dtype=long)
print(m.dtype)

这输出int64；即，它们不是 Python long。

有没有办法创建一个每个元素都是long类型的numpy数组？或者这是 numpy 不支持的一些固定宽度与非固定宽度的问题？如果是这样，是否有任何库(最好有像 numpy 这样的漂亮的 C API)可以做到这一点？

Answer 1

Python 的长整数类型不是原生的 numpy 类型，所以你必须使用 object 数据类型。 object 类型的 numpy 数组的元素可以是任何 python 对象。

例如，

In [1]: x = np.array([1L, 2L, 3L], dtype=object)

In [2]: x
Out[2]: array([1L, 2L, 3L], dtype=object)

In [3]: x[0]
Out[3]: 1L

In [4]: type(x[0])
Out[4]: long

这对你是否有用取决于你想用 long 数组做什么。