Python 使用nltk计算词的搭配

#!/usr/bin/python 
# -*- coding: utf-8 -*-

"""
Created on 2015-1-21
@author: beyondzhou
@name: nltk_compute_collocation.py
"""

import json
import nltk

# Load in human language data
DATA = r"E:eclipseGoogledFile107033731246200681024.json"
data = json.loads(open(DATA).read())

# Number of collocations to find
N = 25

all_tokens = [token for activity in data for token in activity["object"]["content"].lower().split()]

finder = nltk.BigramCollocationFinder.from_words(all_tokens)
finder.apply_freq_filter(2)
finder.apply_word_filter(lambda w: w in nltk.corpus.stopwords.words("english"))
scorer = nltk.metrics.BigramAssocMeasures.jaccard
collocations = finder.nbest(scorer, N)
for collocation in collocations:
    c = " ".join(collocation)
    print c

cabo pulmo
maker faire
bay mini
east bay
pulmo sunrise
never search
mini maker
money supply
desired outcomes
open data
, +
important piece
it"s also
new services
new york