numpy.ndarray中數據轉為int型

首先了解內容與類型

>>>print(a)
(array([[0.01124722],
       [0.21752586],
       [0.05586815],
       [0.03558792]]), array([[ 327],
       [ 366],
       [1887],
       [1153],
       [1792]], dtype=int64))
>>>print(type(a))
<class 'tuple'>
>>>print(a[1])
[[ 327]
 [ 366]
 [1887]
 [1153]
 [1792]]
>>>print(type(a[1]))
<class 'numpy.ndarray'>

接下來可以遍歷a[1]，對每個元素用list()和tolist()轉int，代碼如下：

for i in a[1]：
? ? b = list(i)[0].tolist()

先用list()轉為列表類型，[0]表示獲取第一個值（其實也只有一個值），此時類型為numpy.int64，然后用tolist()便轉為int型。?

numpy.ndarray中數據轉為int型

出現錯誤only size-1 arrays can be converted to Python scalars

經過分析后發現是數據類型出現的錯誤。

原數據為 [63]，使用type得到數據類型為：<class 'numpy.ndarray'>

故將numpy的ndarray類型轉為int (或者你需要的類型，此處為int）

使用 list(m)[0].tolist() 的方法得到如下的結果

63

<class 'int'>

Ok，問題成功解決。