I\'d like to multiply an int16
array but a float
array, with auto rounding, but this fails :
import numpy
A = numpy.array([1, 2, 3
You could use broadcasting to multiply the two arrays and take only the integer part as follows:
In [2]: (A*B).astype(int)
Out[2]: array([ 0, 4, 9, 16])
Timing Constraints:
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
2 ways to solve this:
You can solve this by replacing
A *= B
with
A = (A * B)
or with
numpy.multiply(A, B, out=A, casting='unsafe')
import numpy as np
A = np.float_(A)
A *= B
try this. I think are different array type you get fail.
Cast