Python 字典到 NumPy 的轉換

Question

我正在將 Python 字典轉換為 NumPy 數組。

import numpy as np

myDict = {
        1: "AAA",
        2: "BBB",
        3: "CCC",
        4: "AAA",
        5: "BBB"
        }

myNumPyArray = np.asarray(myDict, dtype=dict, order="C")

print(myNumPyArray) 

輸出

{1: 'AAA', 2: 'BBB', 3: 'CCC', 4: 'AAA', 5: 'BBB'}

為什么它在輸出中顯示字典？

Answer 1

這“失敗”是因為創建了一個包含單個“對象”類型標量值的數組。 這個標量值將是一個指向字典的指針。 如果要將字典的內容轉換為 ndarray，可以使用具有 2 個字段的結構化數組。 一個用於整數索引，另一個用於作為固定長度字符串的值。 使用 np.fromiter 是一種快速方法，因為它避免了創建任何臨時對象。

np.fromiter(myDict.items(), dtype='i4,U3', count=len(myDict))