python - 在 numpy 中禁用静默转换

Question

有没有办法在 numpy 中禁用静默转换？

import numpy as np
a = np.empty(10, int)
a[2] = 4     # OK
a[3] = 4.9   # Will silently convert to 4, but I would prefer a TypeError
a[4] = 4j    # TypeError: can't convert complex to long

numpy.ndarray 对象是否可以配置为在分配任何不是 ndarray 类型的 isinstance() 的值时返回 TypeError？如果不是，最好的替代方法是子类化 numpy.ndarray(并覆盖 __setattr__ 或 __setitem__)吗？

Answer 1

不幸的是 numpy 在数组创建中不提供此功能，您可以设置是否仅在转换数组时才允许转换(查看 numpy.ndarray.astype 的文档)。

您可以使用该功能或​​子类 numpy.ndarray，但也可以考虑使用 array python 本身提供的用于创建类型化数组的模块:

from array import array

a = array('i', [0] * 10)
a[2] = 4                   # OK
a[3] = 4.9                 # TypeError: integer argument expected, got float