更改numpy数组的数据类型

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))

结果如下：