python 초보자입니다. 파싱질문드립니다.

cadava의 이미지

간단한 건데, 중간 중간 맞게 파싱되지 않아서 고민하다가 질문합니다.
부탁드립니다.

open file = fin.csv 에는

abc 1
bbb -1
abc 1.5
ddd 1.1
abc -3
bbb 4
....

fout write file = fout.csv 에는

abc -3
bbb 4
ddd 1.1
....

요렇게 하고 싶습니다.

즉 fin파일에서 1) 첫째 column이 동일하면, 2) 둘째 column의 절대값을 비교하여
3) 큰 값을 갖는 첫째와 둘째 column을 write하는 것입니다.
4) 이때 첫째 column 이 겹치는 값이 없으면 (예로 ddd,) 이것도 column값을 write합니다.

제가 짠 거에선 이상하게 막 중간값이 써져서요...
python으로 코드부탁드립니다.
감사합니다.

익명 사용자의 이미지

import itertools
from operator import itemgetter
 
src = [['abc', 1], ['bbb', -1], ['abc', 1.5], ['ddd', 1.1], ['abc', -3], ['bbb', 4]]
src.sort(key=itemgetter(0))    # abc bbb ddd 순으로 정렬
groups = itertools.groupby(src, itemgetter(0))    # abc, bbb, ddd 을 키로해서 그룹핑
[[k, sorted([d[1] for d in v], key=abs)[-1]] for k, v in groups]    # 그룹된 결과에서 절대값으로 소팅후 가장 마지막 요소 출력
chanik의 이미지

#! /usr/bin/python
# -*- coding: utf-8 -*-
 
fi = open("fin.csv", "r")
fo = open("fout.csv", "w")
 
amax = {}
 
while True:
    line = fi.readline()
    if not line:
        break
    try:
        key, val = line.split()
        if key not in amax or abs(float(val)) > abs(float(amax[key])):
            amax[key] = val
    except:
        continue
 
for key in iter(amax):
    fo.write("%s\t%s\n" % (key, amax[key]))
chanik의 이미지

http://ask.python.kr/questions/

그리고, 파이썬 질문은 위 사이트에서 하시면 응답이 빠를 것입니다.