numpy.ndarray中數(shù)據(jù)轉(zhuǎn)為int型

首先了解內(nèi)容與類型

>>>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]，對(duì)每個(gè)元素用list()和tolist()轉(zhuǎn)int，代碼如下：

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

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

numpy.ndarray中數(shù)據(jù)轉(zhuǎn)為int型

出現(xiàn)錯(cuò)誤only size-1 arrays can be converted to Python scalars

經(jīng)過分析后發(fā)現(xiàn)是數(shù)據(jù)類型出現(xiàn)的錯(cuò)誤。

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

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

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

63

<class 'int'>

Ok，問題成功解決。