方法一:
pred = np.zeros_like(result, dtype=np.uint8)
pred[result > self.seg_conf_th] = 255
找到矩阵resutl中,所有满足
result > self.seg_conf_th
并替换值成255
方法二:
index = np.argwhere(arr < 5)
方法三:
这个就叼了,列表推导式+三元表达式
这个处理是list, 其他类型可以先转换成list(比如
数组转List: defect_words_index_array.tolist()
),再用下面的替换,万能方法
['B' if value == 10 else value for value in class_result_list] # 将class_result_list中的10替换成字母‘B’
注:如果if加到for后面,则是刷选条件,所以后面不能加上else了。
列表推导式其他操作
将字符串'19234BBA'---> 列表‘1’,‘9’,...
[str(real_date[i]) for i in range(len(real_date))]
双循环列表推导式,六的起飞!
>>> [m + n for m in 'ABC' for n in 'XYZ']
['AX', 'AY', 'AZ', 'BX', 'BY', 'BZ', 'CX', 'CY', 'CZ']