python - 如何在python中以键值对的形式打印数组？

Question

我有一个数组集合，例如

 a = [array([0, 1, 2, 3, 4]) array([0, 1, 2, 3]) array([0, 1, 2, 3, 4])
     array([0, 1, 2, 3, 4, 5, 6, 7, 8]) array([0, 2, 3, 4, 5, 8, 9])

有没有办法以键值对的形式显示这个结果

[[(0),(1,2,3,4)],[(1),(0,2,3)],[(2),(0,1,3,4)],[(3),(0,1,2,4,5,6,7,8)]
[(4),(0,2,3,5,8,9)]]

i 将在键中增加 1，并且该值将不包含在值列表中

<小时/>

我尝试了这样的方法，但无法将其放入所需的形式。

c = [id for id in b if id != i] for i, b in enumerate(a)]

Answer 1

使用 enumerate 进行列表理解是一种方法。请注意每个单项键后面的逗号。这表示对象类型是长度为 1 的元组。

from numpy import array

a = [array([0, 1, 2, 3, 4]), array([0, 1, 2, 3]), array([0, 1, 2, 3, 4]),
     array([0, 1, 2, 3, 4, 5, 6, 7, 8]), array([0, 2, 3, 4, 5, 8, 9])]

res = [[(i,), tuple(j for j in arr if j != i)] for i, arr in enumerate(a)]

# [[(0,), (1, 2, 3, 4)],
#  [(1,), (0, 2, 3)],
#  [(2,), (0, 1, 3, 4)],
#  [(3,), (0, 1, 2, 4, 5, 6, 7, 8)],
#  [(4,), (0, 2, 3, 4, 8, 9)]]

或者，您可以创建一个字典:

res_dict = {i: tuple(j for j in arr if j != i) for i, arr in enumerate(a)}

# {0: (1, 2, 3, 4),
#  1: (0, 2, 3),
#  2: (0, 1, 3, 4),
#  3: (0, 1, 2, 4, 5, 6, 7, 8),
#  4: (0, 2, 3, 4, 8, 9)}