如何以数组形式访问组合的元素？

Question

我试图以数组的形式访问组合列表的元素，以便能够操纵组合的结果。a = 1b = 2c = 3d = 4comb = combinations_with_replacement([a, b, c, d], 2)for i in list(comb):     print(i) 我有这段代码，a, b, c, d变量不会是那些特定值。这将返回：(1, 1)(1, 2)(1, 3)(1, 4)(2, 2)(2, 3)(2, 4)(3, 3)(3, 4)(4, 4)我希望能够以数组的形式访问每个组合来操作其元素，我该怎么做？

Answer 1

写一个包装器。

def wrapper_combinations_with_replacement(iterable, r):

    comb = combinations_with_replacement(iterable, r)

    for item in comb:

        yield list(item)

现在你有了一个列表的列表。

a = 1

b = 2

c = 3

d = 4

comb = wrapper_combinations_with_replacement([a, b, c, d], 2)

for i in list(comb): 

    print(i)

结果是：

[1, 1]

[1, 2]

[1, 3]

[1, 4]

[2, 2]

[2, 3]

[2, 4]

[3, 3]

[3, 4]

[4, 4]

或者使用list

list(wrapper_combinations_with_replacement([a, b, c, d], 2))

结果：

[[1, 1],

 [1, 2],

 [1, 3],

 [1, 4],

 [2, 2],

 [2, 3],

 [2, 4],

 [3, 3],

 [3, 4],

 [4, 4]]

Answer 2

comb_list = [list(combination) for combination in (list(comb))]

这将为您提供一个数组形式的组合列表

for array_combination in comb_list:

    print(array_combination)

Output:

> [1, 1]

  [1, 2]

  [1, 3]

  [1, 4]

  [2, 2]

  [2, 3]

  [2, 4]

  [3, 3]

  [3, 4]

  [4, 4]