Python:使用float比较数组元素,得到一个布尔列表

时间:2018-04-24 18:41:41

标签: python list numpy compare

我想比较二维列表的所有元素 带浮点数的数组 X。 结果应该是一个列表: b = [[True,False,...],...]。 我这样试过:

import numpy as np

array = [[a1,a2,...], [a3,a4,...],...,] 
x = 2.0

b = np.array([a >= x for a in array])`

"TypeError: '>=' not supported between instances of 'list' and 'float'"

当我使用一维列表时,它可以正常工作。

提前致谢!!

1 个答案:

答案 0 :(得分:1)

b = np.array([[a >= x for a in row] for row in array])