numpy数组中的列表理解

时间:2019-06-26 12:02:19

标签: python numpy list-comprehension

我有一个类似[[0.50154066 0.49899566 0.50213783 ...]]的numpy数组

我如何使用列表理解来创建一个新数组,该列表使用0.5作为阈值分类为0/1

类似

for(i in array):
 if(i > 0.5):
  newarray.append(1)
 else:
  newarray.append(0)

0 个答案:

没有答案