如何删除列表中的子列表元素?调用list.index()返回结果异常如何解决?
问题原因及修复方案
核心错误:list.remove()方法用法错误
Python列表的remove()方法接收的参数是你要删除的元素值本身,而不是元素的下标索引。你现在把获取到的下标index_det传给remove(),会导致程序尝试在detectedlines列表中查找和index_det值相等的元素(也就是你返回的整数0),而你的列表里存的都是坐标数组,根本不存在这个值,逻辑完全错误。
如果要通过下标删除元素,应该使用pop()方法,把对应代码改成:
detectedlines.pop(index_det)
其他可能的隐藏问题
如果你的detectedlines列表里存在多个完全相同的坐标数组,list.index()方法只会返回第一个匹配到的元素下标,如果你要删除的是后续计算得到的IOU更高的重复坐标,会出现删错元素的问题。这种情况下建议你在遍历检测框的时候直接记录下标,而不是记录元素值再找下标,优化写法如下:
#Loop for each ground truth object compare with all the detections for obj in groundtruth: print('\t',obj) max_iou = 0 # 初始化存储最佳匹配的下标 best_det_idx = 0 # 遍历的时候同时取下标和元素 for idx, det in enumerate(detectedlines): print('\t>',det) iou = calc_iou(obj, det, img_width, img_height) if iou > max_iou: max_iou = iou best_det_idx = idx # 直接通过记录的下标删除元素,不需要再调用index方法 print(best_det_idx) print(detectedlines[best_det_idx]) print(detectedlines) detectedlines.pop(best_det_idx) print(detectedlines)
内容的提问来源于stack exchange,提问作者diogofd8
相关产品推荐
相关产品推荐

