將字典更改為標簽數組

Question

所以我有一個二維數組

X = [[ 7.3571296   0.49626   ]
 [-0.7747436   3.14599   ]
 [ 3.7817762   4.1808457 ]
 [ 4.5332413   6.8228664 ]
 [ 7.4655724  -0.11392868]
 [ 2.416418    4.692072  ]]

和一個簇標簽數組。

y = [1 3 2 2 1 3]

然后我有一個算法可以預測二維數組的標簽。

Z = {1: array([[ 7.3571296   0.49626   ],
               [ 7.4655724  -0.11392868]]), 
     2: array([[ 3.7817762   4.1808457 ]
               [ 2.416418    4.692072  ]]),
     3: array([[-0.7747436   3.14599   ],
               [ 4.5332413   6.8228664 ]])} 

我想將我的預測標簽與原始標簽相匹配，以了解我的算法的准確性。 但是如何將字典格式提取為標簽數組格式？ （即 y_pred = [1 3 2 3 1 2]）

Answer 1

您可以使用字典的keys()方法並將其轉換為list 。

import numpy as np

Z = {1: np.asarray([[7.3571296, 0.49626], [7.4655724, 0.11392868]]),
     2: np.asarray([[3.7817762, 4.1808457], [2.416418, 4.692072]]),
     3: np.asarray([[-0.7747436, 3.14599], [4.5332413, 6.8228664]])}

print(list(Z.keys())) #[1, 2, 3]