如何预先确定numpy数组操作的结果类型?

时间:2017-04-12 15:06:17

标签: python numpy

为了为某些结果创建一个空数组,在基于另外两个数组进行操作时,我需要知道某个操作的结果dtype(例如乘法)。

如何提前确定numpy数组操作的dtype结果?

如果ab是参数数组,我可以通过将值设为零来确定乘法(dtype)的结果*({{ 1}})并进行试运行,如:

0

然而,它似乎有点奖励......或者我可能以错误的方式做到这一点......

因此,使用接受答案中给出的dtype=(a.dtype.type(0) * b.dtype.type(0)).dtype ,代码可以是:

result_type

1 个答案:

答案 0 :(得分:2)

使用numpy.result_type(),在numpy> = 1.6.0

https://docs.scipy.org/doc/numpy/reference/generated/numpy.result_type.html