python - Itertools 与索引的组合

Question

关闭。此题需要 details or clarity 。目前不接受答案。

---

想要改进这个问题吗？通过 editing this post 添加详细信息并澄清问题.

已关闭 6 年前。

Improve this question

假设我有dlist=[[1, 2], [3, 4], [5, 6], [7, 8],[9,10]] 。我想将两个列表及其索引组合在一起。

期望的输出是

[[0,1],[[1,2],[3,4]]], [[0,2],[[1,2],[5,6]]] and so on..

以下代码创建

clist.append([list(itertools.combinations(list(enumerate(dlist)), 2))])


[((0, [1, 2]), (1, [3, 4])), ((0, [1, 2]), (2, [5, 6])),....

相反..

Answer 1

你就快到了。只需将您的 clist 转换为所需的格式即可。itertools.combinations( enumerate(dlist), 2) 将返回格式为 ((index_number_of_sub_arr1, sub_arr1), (index_number_of_sub_arr2, sub_arr2)) 的组合迭代器。您只需遍历迭代器，将其转换为您的格式[[index_number_of_sub_arr1, index_number_of_sub_arr2],[sub_arr1, sub_arr2]]。

# this is like your current clist (iterator)
tmp_list = itertools.combinations( enumerate(dlist), 2) 

# convertion to desired format
clist = [ [[idx1, idx2], [arr1,arr2]] for ((idx1, arr1),(idx2,arr2)) in tmp_list ]