乗算numpyのint型とfloat型の配列

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 

私が手：

TypeError: Cannot cast ufunc multiply output from dtype('float64') to dtype('int16') with casting rule 'same_kind'

Answer 1

あなたは可能性がありbroadcastingを使用して、2つの配列を掛け、整数部分のみを次のように使用します。

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 

Answer 2

import numpy as np 

A = np.float_(A) 
A *= B 

はこれを試してみてください。私はあなたが失敗する異なる配列の種類があると思います。これを解決する

キャスト

Answer 3

2つの方法：は

あなたは

A = (A * B) 

またはで

A *= B 

を置き換えることによって、この問題を解決することができます

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