为了为某些结果创建一个空数组,在基于另外两个数组进行操作时,我需要知道某个操作的结果dtype
(例如乘法)。
如何提前确定numpy数组操作的dtype
结果?
如果a
和b
是参数数组,我可以通过将值设为零来确定乘法(dtype
)的结果*
({{ 1}})并进行试运行,如:
0
然而,它似乎有点奖励......或者我可能以错误的方式做到这一点......
因此,使用接受答案中给出的dtype=(a.dtype.type(0) * b.dtype.type(0)).dtype
,代码可以是:
result_type
答案 0 :(得分:2)
使用numpy.result_type()
,在numpy> = 1.6.0
https://docs.scipy.org/doc/numpy/reference/generated/numpy.result_type.html