Numpy：astype(bool) 和 astype（int）

1 np.array()

将输入转换成数组。

栗子：

import numpy as np

a=[[1,2,1],[2,3,5]]
b=[[0,0,0],[2,3,5]]

c=np.array(a)
d=np.array(b)

print(c)
print(d)

# result

[[1 2 1]
 [2 3 5]]

[[0 0 0]
 [2 3 5]]

简单的把list列表转化为数组。

2 type(bool)

正如astype的中文意思，作为布尔类型，也就是true or false。

栗子：

import numpy as np

a=[[1,2,1],[2,3,5]]
b=[[0,0,0],[2,3,5]]

c=np.array(a).astype(bool)
d=np.array(b).astype(bool)

print(c)
print(d)

结果：

[[ True True True]
 [ True True True]]

[[False False False]
 [ True True True]]

0代表False 非0代表True。

3 astype（int）

就是转化为整型数据。True和False转化为整型数据是什么样子呢？不就是0和1嘛！

栗子：

import numpy as np

a=[[1,2,1],[2,3,5]]
b=[[0,0,0],[2,3,5]]

c=np.array(a).astype(bool).astype(int)
d=np.array(b).astype(bool).astype(int)

print(c)
print(d)

结果：

[[1 1 1]
 [1 1 1]]

[[0 0 0]
 [1 1 1]]

参考：

• https://blog.csdn.net/wuxulong123/article/details/103387222