创建一个由零组成的numpy数组,该数组的dtype与另一个数组相同,但形状不同



如何创建一个与另一个数组具有相同数据类型的NumPy数组?我知道zeros_like(),但它给出的形状与传递的数组相同。我正在寻找一个不同的形状,但与源数组相同的数据类型。

NumPy数组有一个名为dtype的属性。在创建具有相同数据类型的新数组时,只需使用此属性。

a = np.array([0,1,2])
print(a.dtype) # dtype('int64')
b = np.array([3.1,4.1,5.1], dtype=a.dtype)
print(b) # array([3, 4, 5])

最新更新