gpt4 book ai didi

python - 在numpy中获取结果数组的dtype

转载 作者:太空狗 更新时间:2023-10-30 00:51:57 32 4
gpt4 key购买 nike

我想为数组操作的输出预分配内存,我需要知道它的数据类型。下面我有一个函数可以执行我想要它执行的操作,但是非常丑陋。

import numpy as np

def array_operation(arr1, arr2):
out_shape = arr1.shape
# Get the dtype of the output, these lines are the ones I want to replace.
index1 = ([0],) * arr1.ndim
index2 = ([0],) * arr2.ndim
tmp_arr = arr1[index1] * arr2[index2]
out_dtype = tmp_arr.dtype
# All so I can do the following.
out_arr = np.empty(out_shape, out_dtype)

上面的内容很丑陋。 numpy 是否有执行此操作的函数?

最佳答案

您正在寻找numpy.result_type .

(顺便说一句,您是否意识到可以将所有多维数组作为一维数组访问?您不需要访问 x[0, 0, 0, 0, 0] -- 你可以访问 x.flat[0]。)

关于python - 在numpy中获取结果数组的dtype,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/7285612/

32 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com