根据另一个列表中的元素进行列表划分

Question

我有两个列表如下

a = [646, 650, 654, 658, 662, 666, 670, 674, 678, 682, 686, 690, 694, 698, 702, 706, 13565, 13569, 13573, 13577, 13581, 13585, 13589, 13593, 13597, 13601, 13605, 13609, 13613, 13617, 13621, 13625, 13629, 13633, 13637, 13641, 13645, 13649, 13653, 13657, 13661, 21237, 21241, 21245, 21249, 21253, 21257, 21261, 21265, 21269, 21273, 21277, 21281, 21285, 21289, 21293, 21297, 21301, 21305, 21309, 21313, 21317, 21321, 21325, 21329, 21333, 21337, 21341, 21345]

b = [646, 706, 13661, 21345]

所以基本上我想根据列表b 中的开始停止值将列表a 分成更小的块。例如。 像这样的

[
[646, 650, 654, 658, 662, 666, 670, 674, 678, 682, 686, 690, 694, 698, 702, 706],
[13565, 13569, 13573, 13577, 13581, 13585, 13589, 13593, 13597, 13601, 13605, 13609, 13613, 13617, 13621, 13625, 13629, 13633, 13637, 13641, 13645, 13649, 13653, 13657, 13661],
[21237, 21241, 21245, 21249, 21253, 21257, 21261, 21265, 21269, 21273, 21277, 21281, 21285, 21289, 21293, 21297, 21301, 21305, 21309, 21313, 21317, 21321, 21325, 21329, 21333, 21337, 21341, 21345]
]

有人可以帮我解决这个问题吗？

Answer 1

解决方案 1：使用二分法

我将通过使用bisect 模块来查找a 中的每个项目将在b 中插入的位置来确定项目属于哪个bin 来解决这个问题。

此解决方案不需要对a 进行排序，但它确实要求对b 进行排序。

bin_boundaries = sorted(b)
results = [[] for _ in range(len(bin_boundaries)+1)]
for i in a:
    pos = bisect.bisect_left(bin_boundaries, i)
    results[pos].append(i)
print(results)

现在，您没有指定是否需要与上一个或下一个 bin 中的边界相等的项目。我把它放在以前的箱子里。如果您指的是下一个，请将上面的 bisect_left 替换为 bisect_right。

我还输出了您预期的输出显示的另外两个 bin：第一个 bin 的项目小于第一个 bin 边界，最后一个 bin 项目大于最后一个边界。如果您想删除这些边缘箱，请添加 results = results[1:-1]。

解决方案 2：只需遍历每个 bin 的列表

现在，这是一个更简单的解决方案，它只为每个 bin 遍历 a：

bin_boundaries = sorted(b)
results = []
for low, high in zip(bin_boundaries[:-1], bin_boundaries[1:]):
    results.append([i for i in a if i > low and i <= high])
print(results)

这一次，我没有创建边缘箱。同样，修复 > 和 <= 以匹配您在边缘处实际需要的语义。

这个外层循环也可以变成一个列表推导，给你这个嵌套列表推导和一个非常紧凑的解决方案：

results = [
    [i for i in a if i > low and i <= high]
    for low, high in zip(bin_boundaries[:-1], bin_boundaries[1:])
]

Answer 2

从您的示例中，我了解到您希望第一个区间包含边界 ([646, 706])，而其他区间必须仅包含上边界 (]706, 13661], ]13661, 21345)。

我在这里使用 .index 方法和一个考虑第一个区间的下边界并为其他区间排除它的 for 循环：

lists_result = []

for i in range(len(b[:-1])):
    idx_inf = a.index(b[i])
    idx_sup = a.index(b[i+1])
    if i == 0:
        lists_result.append(a[idx_inf:idx_sup+1])
    else:
        lists_result.append(a[idx_inf+1:idx_sup+1])