要自定義數據類型,可以使用np.dtype
函數并傳入一個字典來定義數據類型的字段和其對應的數據類型。例如,以下示例定義了一個名為person
的自定義數據類型,包含了姓名(字符串類型)、年齡(整數類型)和身高(浮點數類型)三個字段:
import numpy as np
# 定義自定義數據類型
person_dtype = np.dtype({
'names': ['name', 'age', 'height'],
'formats': ['S10', np.int32, np.float64]
})
# 創建一個包含自定義數據類型的NumPy數組
data = np.array([('Alice', 25, 1.65), ('Bob', 30, 1.75)], dtype=person_dtype)
print(data)
輸出結果為:
[(b'Alice', 25, 1.65) (b'Bob', 30, 1.75)]
在這個示例中,我們定義了一個包含三個字段的自定義數據類型person_dtype
,然后使用該數據類型創建了一個包含兩條數據的NumPy數組data
。