a.astype(数据类型) 后面选择自己需要的数据类型即可
import numpy as np
a=np.random.random((5,4,3))
b=np.random.randint(1,3,(5,4,3))
a=a+b
print(a)
print(a.astype(np.int16))
结果如下:
a.astype(数据类型) 后面选择自己需要的数据类型即可
import numpy as np
a=np.random.random((5,4,3))
b=np.random.randint(1,3,(5,4,3))
a=a+b
print(a)
print(a.astype(np.int16))
结果如下: