Numpy 无法从 dtype 转换 ufunc 乘法输出

Question

我想将一个 int16 float 相乘，自动舍入，但这失败了：

 import numpy

A = numpy.array([1, 2, 3, 4], dtype=numpy.int16)
B = numpy.array([0.5, 2.1, 3, 4], dtype=numpy.float64)

A *= B

我得到：

类型错误：无法使用转换规则“same_kind”将 dtype(‘float64’) 的 ufunc 乘法输出转换为 dtype(‘int16’)

原文由 Basj 发布，翻译遵循 CC BY-SA 4.0 许可协议

Answer 1

解决这个问题的2种方法：

您可以 通过更换 来解决这个问题

A *= B

和

A = (A * B)

或 与

numpy.multiply(A, B, out=A, casting='unsafe')

原文由 seralouk 发布，翻译遵循 CC BY-SA 4.0 许可协议

Answer 2

您可以使用 broadcasting 将两个数组相乘并只取整数部分，如下所示：

 In [2]: (A*B).astype(int)
Out[2]: array([ 0,  4,  9, 16])

时间限制：

 In [8]: %timeit (A*B).astype(int)
1000000 loops, best of 3: 1.65 µs per loop

In [9]: %timeit np.multiply(A, B, out=A, casting='unsafe')
100000 loops, best of 3: 2.01 µs per loop

原文由 Nickil Maveli 发布，翻译遵循 CC BY-SA 3.0 许可协议