新人求教

tingshuibanxia

向各位大神请教一个排序问题！
文本包含3列，tab分隔

A1    X    XX “X1”; XXX "XX1";
A1    Y    YY “YY”; YYY "YYY"; Y_N "1"; E
A1    Y   YY “YY”; YYY "YYY"; Y_N "3"; E
A1    Y    YY “YY”; YYY "YYY"; Y_N "2"; E
A1    Z    ZZ "ZZ"; ZZZ "ZZZ";
A1    X    XX “X2”; XXX "XX2";
A1    Y   YY “YY”; YYY "YYY"; Y_N "3"; E
A1    Z    ZZ "ZZ"; ZZZ "ZZZ";
A1    Y    YY “YY”; YYY "YYY"; Y_N "2"; E
A1    Y    YY “YY”; YYY "YYY"; Y_N "1"; E
A1    Y    YY “YY”; YYY "YYY"; Y_N "4"; E
B1    X    XX “X4”; XXX "XX4";
B1    Z    ZZ "ZZ"; ZZZ "ZZZ";
B1    Y    YY “YY”; YYY "YYY"; Y_N "2"; E
B1    Y    YY “YY”; YYY "YYY"; Y_N "1"; E



我想把第2列按X Z Y 排序，同时Y的几行按照最后的数字进行排序。
每次X第一次出现后，下跟的几行数目不定，到下一个X出现，最后仍按照现在的模式输出如下，还请大神指点！
A1    X    XX “X1”; XXX "XX1";
A1    Z    ZZ "ZZ"; ZZZ "ZZZ";
A1    Y    YY “YY”; YYY "YYY"; Y_N "1"; E
A1    Y    YY “YY”; YYY "YYY"; Y_N "2"; E
A1    Y   YY “YY”; YYY "YYY"; Y_N "3"; E
A1    X    XX “X2”; XXX "XX2";
A1    Z    ZZ "ZZ"; ZZZ "ZZZ";
A1    Y    YY “YY”; YYY "YYY"; Y_N "1"; E
A1    Y    YY “YY”; YYY "YYY"; Y_N "2"; E
A1    Y   YY “YY”; YYY "YYY"; Y_N "3"; E
A1    Y    YY “YY”; YYY "YYY"; Y_N "4"; E
B1    X    XX “X”; XXX "XX";
B1    Z    ZZ "ZZ"; ZZZ "ZZZ";
B1    Y    YY “YY”; YYY "YYY"; Y_N "1"; E
B1    Y    YY “YY”; YYY "YYY"; Y_N "2"; E

haooooaaa

1. m = {"Y":[], "Z":[]}
   
2. 
   
3. def pp(n):
   
4.     for j in sorted(n):
   
5.         print j
   
6.         n.remove(j)
   
7. 
   
8. with open("tt") as f:
   
9.     for i in f:
   
10.         tmp = i.split("\t")
    
11.         if tmp[1] == "X":
    
12.             if len(m["Y"]):
    
13.                 pp(m["Z"])
    
14.                 pp(m["Y"])
    
15.             print i.strip()
    
16.         else:
    
17.             m[tmp[1]].append(i.strip())
    
18. 
    
19.     pp(m["Z"])
    
20.     pp(m["Y"])

复制代码

zxy877298415

回复 1# tingshuibanxia

1. a={}
   
2. b={}
   
3. c=[]
   
4. import re
   
5. with open('file.txt','r') as f:
   
6.         for l in f:
   
7.                 if re.search(r' +X +',l):
   
8.                         a.setdefault(l,[])
   
9.                         b.setdefault(l,[])
   
10.                         c.append(l)
    
11.                         s = l
    
12.                 elif re.search(r' +Y +',l):
    
13.                         a[s].append(l)
    
14.                 elif re.search(r' +Z +',l):
    
15.                         b[s].append(l)
    
16. 
    
17. for i in sorted(c,key=lambda x:int(re.search(r'(?<=X)\d+',x).group(0))):
    
18.         print i,"".join(b[i]),"".join(sorted(a[i],key=lambda x:int(re.search(r'(?<=\")\d+(?=\")',x).group(0)))),

复制代码

MMMIX

回复 1# tingshuibanxia

1. #!/usr/bin/env python3
   
2. 
   
3. import sys, re
   
4. 
   
5. def file2block(filename):
   
6.     with open(filename) as file:
   
7.         block = { 'X':None, 'Y':[], 'Z':[] }
   
8.         for line in file:
   
9.             s = re.search('\t([XYZ])\t', line).group(1)
   
10.             if s == 'X':
    
11.                 if block['X']:
    
12.                     yield block
    
13.                     block = { 'X':line, 'Y':[], 'Z':[] }
    
14.                 else:
    
15.                     block['X'] = line
    
16.             else:
    
17.                 block[s].append(line)
    
18. 
    
19.         yield block
    
20. 
    
21. 
    
22. def print_block(block):
    
23.     print(block['X'], end='')
    
24. 
    
25.     for Z in block['Z']:
    
26.         print(Z, end='')
    
27. 
    
28.     # Schwartzian transform
    
29.     # Note: May not be worth it.
    
30.     Ys = map(lambda r: r[0],
    
31.              sorted(map(lambda s: [s, re.search('"(\d+)"', s).group(1)], block['Y']),
    
32.                     key = lambda r: r[1]))
    
33. 
    
34.     for Y in Ys:
    
35.         print(Y, end='')
    
36. 
    
37. 
    
38. def main():
    
39.     for b in file2block(sys.argv[1]):
    
40.         print_block(b)
    
41. 
    
42. if __name__ == '__main__':
    
43.     main()
    
44. 
    

复制代码