Integer division operation yields float

[cod3monk]

import sympy
import numpy

i = sympy.var('i', integer=True)

# lambdify function with integer division (//)
f = i // 3
lf = sympy.lambdify(i, f, modules='numpy')

# map to array of integers
lf(numpy.arange(0, 10, dtype='int64'))

while the integer division is correctly done with floor and /, this gives an array of floats (array([0., 0., 0., 1., 1., 1., 2., 2., 2., 3.])), but I would expect an array of integers (array([0, 0, 0, 1, 1, 1, 2, 2, 2, 3])) as i is an integer, // gives an integer and the input datatype is also an integer. Is there anyway to force returned types?

[Simardeep27]

I can confirm this issue.

If you want to force the returned type, you can use numpy.vectorize(sympy.lambdify(i,f, modules='numpy'),otypes=[int]), you can set the output data type using otype parameter.

You can view this link for np.vectorize : ( https://numpy.org/doc/stable/reference/generated/numpy.vectorize.html )

This issue was raise earlier too , you can refer to #5642