我有两个嵌套列表,形式:
a = [[1,2,3,4,5],[3,4,5,6,7,8,9],[5],[1,2,3,6,7,8,9]]
b = [[1,4],[6,9]]
我想要一个将接收这两个列表的函数,使用列表b
的每个子列表中的值,并使用该函数删除列表a
中的值,以便输出看起来像这样:>
[[5],[5],[5],[]]
即函数应该迭代列表a
中的所有值,并删除诸如1 <= value <= 4
和6 <= value <= 9
目前我有这个:
def remove_unnecessary_peaks(peaks_total,peaks_to_be_excluded):
redacted_peak_list = peaks_total.copy()
i=0
for item in peaks_total:
for value_set in peaks_to_be_excluded:
for peak in item:
#print (value_set[0],peak,value_set[1])
if value_set[0]<=peak<=value_set[1]:
redacted_peak_list[i].remove(peak)
i=i+1
return (redacted_peak_list)
a = [[1,2,3,4,5],[3,4,5,6,7,8,9],[5],[1,2,3,6,7,8,9]]
b = [[1,4],[6,9]]
print(remove_unnecessary_peaks(a,b))
我已经通过执行redacted_peak_list = peaks_total.copy()
复制了数据,所以我不修改peaks_total
变量,因为它会破坏索引编制。在Spyder的调试模式下,我看到变量peak
未被正确索引,我看到它跳过了子列表a
中的数字。
我的输出是[[2, 4, 5], [4, 5, 7, 9], [5], [2, 7, 9]]
,而我想要[[5],[5],[5],[]]
答案 0 :(得分:3)
您可以像这样使用列表理解:
[[i for i in s if not any(l <= i <= h for l, h in b)] for s in a]
这将返回:
[[5], [5], [5], []]