假设我有这样一个序列列表。
我想要移除所有总和等于N的序列和/或包含连续子数组和为N的序列。
例如,如果N = 4,则(1,1,2)无效,因为它们的总和是4。(1,1,3)也不合法,因为(1,3)的和也是4。(1,3,1)由于相同的原因也无效。
lst = [
(1,1,1), (1,1,2), (1,1,3),
(1,2,1), (1,2,2), (1,2,3),
(1,3,1), (1,3,2), (1,3,3),
(2,1,1), (2,1,2), (2,1,3),
(2,2,1), (2,2,2), (2,2,3),
(2,3,1), (2,3,2), (2,3,3),
(3,1,1), (3,1,2), (3,1,3),
(3,2,1), (3,2,2), (3,2,3),
(3,3,1), (3,3,2), (3,3,3)
]
有什么方法可以做到这一点吗?
我目前正在尝试查看是否能够删除总和不一定是N的倍数但不删除其连续子数组的序列,但我一直没有成功。
for elements in list(product(range(1,n), repeat=n-1)):
lst.append(elements)
for val in lst:
if np.cumsum(val).any() %n != 0:
lst2.append(val) # append value to a filtered list