python 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型。