如何对字符串列表中的组件进行分类

Question

我有一个字符串列表，需要遍历这些字符串并对组件进行分类。组件在列表中的类别，但我不知道从哪里开始

lst = [
'CAPACITOR
  C500 1u PN"1";
  C501 1u PN"2";

RESISTOR
  R100 10 PN"10";
  R101 10 PN"11";'

,

'CAPACITOR
  C500 1u PN"1";
  C507 1u PN"2";

CONNECTOR
  TP100 NT;
  TP101 NT;']

#Above formatted as it is easier to see

lst = [ 'CAPACITOR\n  C500 1u PN"1";\n  C501 1u PN"2";\n\nRESISTOR\n  R100 10 PN"10";\n  R101 10 PN"11";\n'  ,  '\nCAPACITOR\n  C500 1u PN"1";\n  C507 1u PN"2";\n\nCONNECTOR\n  TP100 NT;\n  TP101 NT;']

我期望得到的结果是

C500 , Capacitor
C501 , Capacitor
R100 , Resistor
R101 , Resistor
C507 , Capacitor
TP100 , Connector
TP101 , Connector

C500只需要定义一次

我能够得到一个列表，只是组件，但不知道如何得到他们在什么类别

Answer 1

假设格式正是显示的格式，它只是一个拆分序列和一个字典：

In [10]: lst = [ 
    ...: '''CAPACITOR 
    ...:   C500 1u PN"1"; 
    ...:   C501 1u PN"2"; 
    ...:  
    ...: RESISTOR 
    ...:   R100 10 PN"10"; 
    ...:   R101 10 PN"11";''', 
    ...:  
    ...: '''CAPACITOR 
    ...:   C500 1u PN"1"; 
    ...:   C507 1u PN"2"; 
    ...:  
    ...: CONNECTOR 
    ...:   TP100 NT; 
    ...:   TP101 NT;''']

In [11]: d = {} 
    ...: for s in lst: 
    ...:     for typelist in s.split('\n\n'): 
    ...:         Type, *elements = typelist.split('\n') 
    ...:         for element in elements: 
    ...:             name = element.strip().split()[0] 
    ...:             d[name] = Type 
    ...:                                                                                  

In [12]: d                                                                                
Out[12]: 
{'C500': 'CAPACITOR',
 'C501': 'CAPACITOR',
 'R100': 'RESISTOR',
 'R101': 'RESISTOR',
 'C507': 'CAPACITOR',
 'TP100': 'CONNECTOR',
 'TP101': 'CONNECTOR'}

Answer 2

这是一种使用Regex的方法

例如：

import re

lst = [ 'CAPACITOR\n C500 1u PN"1";\n C501 1u PN"2";\n\n RESISTOR\n R100 10 PN"10";\n R101 10 PN"11";\n ' , '\n CAPACITOR\n C500 1u PN"1";\n C507 1u PN"2";\n \n CONNECTOR\n TP100 NT; \n TP101 NT;'] 
result = []
for i in lst:
    for j in i.splitlines():
        if j.strip():
            category = re.match(r"\s*(\b[A-Z]+\b)", j)     #Check if line has category  > All Capital letters
            if category:
                result.append([category.group(1)])
            else:
                component = re.match(r"\s+(\b[A-Z0-9]+\b)", j)   #Check if line has component  > Capital letter and int. 
                if component:
                    result[-1].append(component.group(1))                    

for i in result:
    category = i[0]
    for component in i[1:]:
        print(component, category)

输出：

C500 CAPACITOR
C501 CAPACITOR
R100 RESISTOR
R101 RESISTOR
C500 CAPACITOR
C507 CAPACITOR
TP100 CONNECTOR
TP101 CONNECTOR