python2.x sort()をpython3.x用に修正したい。

freq = dict()
for line in open(filename, 'rb'):
    for c in line:
        freq.setdefault(c, 0)
        freq[c] += 1

# 頻度が少ないものから順に枝をまとめてツリー形成
wtree = [i for i in freq.items()]
while len(wtree) > 1:
    # 出現頻度ワースト2の枝を見つけてまとめる
    wtree.sort(lambda a,b: cmp(b[1], a[1]))   //ここをpython3用に変えたい
    b2, b1 = (wtree.pop(), wtree.pop())
    wtree.append(((b1[0], b2[0]), b1[1] + b2[1]))

quickquip

2017/10/24 02:43

ソースコードは、最低限読めるように書いてください。 https://teratail.com/help/question-tips#questionTips3-5-1

Cielwoo

2017/10/24 02:48

初めてなので使い方がよくわかりませんでした。ありがとうございます。

行動規範の内容に同意します

回答1件

ベストアンサー

2と3系ではsortの使い方が異なるようですね。
functools.cmp_to_keyを使うと楽できるようです。

参考：
Python2からPython3.0での変更点
 cmp パラメータを利用した古い方法

Python
1import functools
2freq = {'a':4,'b':3,'c':2,'d':4,'e':2}
3
4# cmpの詳細は分からないので適当に
5def cmp( v1, v2):
6    return v1-v2
7
8# 頻度が少ないものから順に枝をまとめてツリー形成
9wtree = [i for i in freq.items()]
10while len(wtree) > 1:
11    # 出現頻度ワースト2の枝を見つけてまとめる
12    #wtree.sort(lambda a,b: cmp(b[1], a[1]))        # ここをpython3用に変えたい
13    wtree.sort( key=functools.cmp_to_key(lambda a,b: cmp(b[1], a[1])))
14    b2, b1 = (wtree.pop(), wtree.pop())
15    wtree.append(((b1[0], b2[0]), b1[1] + b2[1]))
16
17print(wtree) # [((('d', 'a'), (('c', 'e'), 'b')), 15)]