python，numpy：強制將新數組設置為float類型

Question

我時不時地寫這樣的代碼：

import numpy as np   
a = np.array([1,2,3])
a[1]=3.3
a[2] *= 50
print(a)

在這里，我不inted a被initalized為int ，但float ，但正如我說，我忘了。

現在，有一種方法可以確保除非顯式指定了dtype，否則此類初始化默認為float？

Answer 1

並非沒有改變來源，沒有。 您的選擇是：

養成在數字末尾加點的習慣：

np.array([1.,2.,3.])

明確使用dtype ：

np.array([1,2,3], dtype=float)

新增功能：

def ozi_array(*args, **kwargs):
    if 'dtype' not in kwargs:
        return np.array(*args, dtype=float, **kwargs)
    return np.array(*args, **kwargs)

Answer 2

使用dtype參數，請參見此處 ：

>>> import numpy as np
>>> np.array([1, 2, 3], dtype=float)
array([ 1.,  2.,  3.])