python - 很难？第一列或第二列相同的行合并?

Question

某一行中的字母，如果有在其他行也出现，则将他们合并为一行。例如 {代码...} 结果 {代码...}

巴扎黑 · Answer

也不是说很容易拉，先写一个粗糙的版本(代码又丑顺序又完全不管XD)，再来慢慢改进吧:

char2id = {}
id2charset = {}

def getcharset(c):
    try:
        return id2charset[char2id[c]]
    except:
        return None

def newcharset(chars):
    newset = set(chars)
    return newset

def merge(charset1, charset2):
    if id(charset1)==id(charset2):
        return
    charset1 |= charset
    for c in charset2:
        char2id[c] = id(charset1)
    id2charset.pop(id(charset2))

with open('test2') as reader:
    for line in reader:
        chars = line.strip().split()
        newset = newcharset(chars)
        id2charset[id(newset)] =newset

        for c in chars:
            charset = getcharset(c)
            if charset:
                merge(newset, charset)
            else:
                char2id[c] = id(newset)

with open('report', 'w') as writer:
    for id, charset in id2charset.items():
        print(' '.join(charset), file=writer)

资料 test:

A B
C A
D C
E F
N G
C N
X Y
F P
P Q
X Z

结果:

P E Q F
X Y Z
B D C G N A

建议你检查的 script 这样写就好:

from collections import Counter

with open('report', 'r') as reader:
    ct = Counter()
    for line in reader:
        ct += Counter(line.strip().split())

for item, count in ct.most_common():
    if count <= 1:
        break
    print(item, count)

我回答过的问题: Python-QA

巴扎黑 · Answer