Python从列出的对象列表中提取属性

Question

我有程序集类实例的列表。列表中的程序集总数可以更改。

ASM_list = [ASM1,ASM2,ASM3...etc]

每个组件都有一个零件列表（我已经过滤了所有组件都具有相同的零件）

ASM1.parts = ['hood','fender','door']
ASM2.parts = ['hood','fender','door']

每个部分都有属性

ASM1.parts[0].color = green
ASM2.parts[0].color = blue

我想打印出每个装配体的零件属性。示例输出：

part, ASM1, ASM2, ASM3
hood, green, blue, red
fender, black, red, yellow
door, yellow, green, orange

我在使用 zip 或 map 执行此操作时遇到问题。

我在下面尝试的示例。

parts_list = []
for ASM in ASM_list:
  parts_list.append([p.color for p in ASM.parts])
print zip(*parts_list)

“Python 食谱”和其他几个堆栈溢出帖子显示了如何同时遍历多个列表，我遇到的问题是程序集的数量发生了变化，所以我无法编写：

for a,b,c in ASM_lists:
  a.part.color, b.part.color, c.part.color

谢谢！

Answer 1

假设您有一个 part_names 列表 - 如果您的部分对象对象具有 name 属性以及 color 属性，您可以通过以下方式获得：

part_names = []
for part in ASM_list[0].parts:
    part_names.append(part.name)

然后你可以得到你想要的输出：

headers = ["part"]
for i, ASM in enumerate(ASM_list):
    headers.append("ASM%d" % (i+1))
print(", ".join(headers))

for i, part_colors in enumerate(zip(*parts_list)):
    part_name = part_names[i]
    print(part_name + ", " + ", ".join(part_colors))