代码之家 › 专栏 › 技术社区 › Will

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

key-value list-comprehension list python-3.x python

Will · 技术社区 · 7 年前

我有一组数组,如

 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)]

2 回复 | 直到 7 年前

jpp 7 年前

使用列出理解 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)}

Aaditya Ura 7 年前

您可以尝试numpy方法:

import numpy as np

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

print([[(i,),tuple(np.delete(j,np.argwhere(j==i)))] for i,j 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, 5, 8, 9)]]

推荐文章

Yorick Robinson · 如何从嵌套列表或字典理解中返回我期望的内容?

2 年前

A.E · python中具有多个if语句的列表理解

2 年前

Sachin Sinkar · 对字典列表的列表理解,以分别获取每个键的值

2 年前

Aaryan · Python列表:如果没有前缀,则添加前缀

2 年前

windyvation · 如何将第一个唯一的多重索引设置为0,并计算其他索引的值

2 年前

Semisonic · 如何在列表理解中添加额外的中间步骤?

2 年前

Sun Jar · 用Python填写大小不均的列表

2 年前

Shuvashish Roy · 将字典列表展平为数据帧列

2 年前

Python learner Shaavin · 如何使用列表理解获取列表中的数字总和?

2 年前

MarkS · 列中列表中元素的值计数

2 年前