특정 조건에서 파일에서 줄을 추출하는 방법은 무엇입니까?

특정 조건에서 파일에서 줄을 추출하는 방법은 무엇입니까?

50K 줄의 파일이 있다고 가정하고 (Account,Cusip)을 기본 키로 간주합니다. 동일한 기본 키를 가진 줄 중에서 마지막 줄을 유지하고 싶습니다. 동일한 기본 키를 가진 줄의 경우 마지막 줄을 유지하는 줄 어떻게 해야 합니까?

Account  Cusip   Amount  
A        a       500
A        a       450
A        b       500
B        b       600
B        b       350
C        a       300
C        c       100
C        c       200
C        c       500

예상 결과

Account  Cusip   Amount 
A        a       450
A        b       500
B        b       350
C        a       300
C        c       500

답변1

예상대로 수행하려면 다음 Python 스크립트를 사용하십시오. 파일에는 1.txt입력 데이터가 포함되어 있습니다.

key1=''
key2=''
amount=''
with open('1.txt') as f:
   for line in f:
        [nkey1,nkey2,namount]=line.split()
        if nkey1==key1 and nkey2==key2:
                amount=namount
        else:
                print '%s\t%s\t%s' % (key1,key2,amount)
                key1=nkey1
                key2=nkey2
                amount=namount
   print '%s\t%s\t%s' % (key1,key2,amount)

관련 정보