如何将我的输出组所有相似的数字转换为特定数量的组?

Question

所以我写了这段代码，目的是让代码将列表中的不同数字组合在一起，总共n: int组

编辑如果您不明白代码的目的是什么，请查看注释，我已经在那里解释了。谢谢：)

def calcdifference(lst: list):
for x in lst:
return (x -= x)
print(calcdifference(lst=[4,5,6,4,3,2,3,4,5]))
def grouping(lst: list, n: int):
if calcdifference(x) in list == max(calcdifference(x)):
lst.append(x)
print(grouping(lst=[4,5,6,4,3,2,3,4,5]))

n：int 表示单个列表中允许的组数，因此如果 n 为 3，则数字将分组为 (x,...)、(x,....) (x,...)如果 n = 2，则数字将分组为 (x,..)，(x,...)。

但是，我的代码在 n 个元素的列表中打印出所有可能的组合。但它并没有将数字组合在一起。所以我想要的是：例如，如果输入是

[10,12,45,47,91,98,99]

如果 n = 2，则输出为

[10,12,45,47] [91,98,99]

如果 n = 3，则输出为

[10,12] [45,47] [91,98,99]

我应该对代码进行哪些更改？

注意：请不要使用内置函数或导入，因为我想尽可能少地使用内置函数

重要提示：代码应该能够为提供的每个列表打印n >= len(lst)组合

Answer 1

您可以尝试以下操作：

def grouping(lst, n):
diff = enumerate((abs(x - y) for x, y in zip(lst, lst[1:])), start=1)
cut = sorted(x[0] for x in sorted(diff, reverse=True, key=lambda x: x[1])[:n-1])
cut = [0, *cut, len(lst)] # add 0 and last index
return [lst[i:j] for i, j in zip(cut, cut[1:])] # return slices
lst = [10,12,45,47,91,98,99]
print(grouping(lst, 2))
print(grouping(lst, 3))
print(grouping(lst, 4))

输出：

[[10, 12, 45, 47], [91, 98, 99]]
[[10, 12], [45, 47], [91, 98, 99]]
[[10, 12], [45, 47], [91], [98, 99]]

诚然，它非常复杂，也许不那么蟒蛇。可能有更有效的方法。无论如何，以下是一些解释...

在第一行中，diff是一个(某种)列表，其中包含元组(i, d)，使得lst在第一项和第i-1项和第i项之间具有d的差异。

第二行更复杂。首先，sorted(diff, reverse=True, key=lambda x: x[1])根据第二个元素对这些元组进行排序，即代表最高跳跃的元组排在第一位。

然后sorted(...)[:n-1]选取第一个 n-1 元组。这些将是要使用的n-1削减。

生成器理解(x[0] for x in ...)只选择每个元组的第一项;也就是说，我们不再需要差异。

然后sorted(...)将再次对这些切割位置进行排序，这将使后续行正常工作。

---

如果您出于某种原因不愿意使用lambda(实际上operator.itemgetter(1)比lambda x: x[1]更好)，您可以为此创建自定义函数。

def get_1st(x):
return x[1]
def grouping(lst, n):
diff = enumerate((abs(x - y) for x, y in zip(lst, lst[1:])), start=1)
cut = sorted(x[0] for x in sorted(diff, reverse=True, key=get_1st)[:n-1])
cut = [0, *cut, len(lst)] # add 0 and last index
return [lst[i:j] for i, j in zip(cut, cut[1:])] # return slices