使用pickle将巨大的bigram词典保存到文件


问题内容

我的一个朋友写了这个小程序。的textFile是1.2GB大小(报纸7年价值)。他成功地创建了词典,但是无法使用pickle(程序挂起)将其写入文件。

import sys
import string
import cPickle as pickle

biGramDict = {}

textFile = open(str(sys.argv[1]), 'r')
biGramDictFile = open(str(sys.argv[2]), 'w')


for line in textFile:
   if (line.find('<s>')!=-1):
      old = None
      for line2 in textFile:
         if (line2.find('</s>')!=-1):
            break
         else:
            line2=line2.strip()
            if line2 not in string.punctuation:
               if old != None:
                  if old not in biGramDict:
                     biGramDict[old] = {}
                  if line2 not in biGramDict[old]:
                     biGramDict[old][line2] = 0
                  biGramDict[old][line2]+=1
               old=line2

textFile.close()

print "going to pickle..."    
pickle.dump(biGramDict, biGramDictFile,2)

print "pickle done. now load it..."

biGramDictFile.close()
biGramDictFile = open(str(sys.argv[2]), 'r')

newBiGramDict = pickle.load(biGramDictFile)

提前致谢。

编辑
任何有兴趣的人,我将简要解释该程序的作用。假设您的文件格式大致如下:

<s>
Hello
,
World
!
</s>
<s>
Hello
,
munde
!
</s>
<s>
World
domination
.
</s>
<s>
Total
World
domination
!
</s>
  • <s> 是句子的分隔符。
  • 每行一个字。

会生成一个biGramDictionary,以供以后使用。
像这样的东西:

{
 "Hello": {"World": 1, "munde": 1}, 
 "World": {"domination": 2},
 "Total": {"World": 1},
}

希望这可以帮助。现在,该策略已更改为使用mysql,因为sqlite无法正常工作(可能是因为大小)


问题答案:

Pickle仅用于编写完整的(小型)对象。您的词典甚至太大,甚至无法保存在内存中,最好改用数据库,这样您可以一次而不是一次存储和检索条目。

您可以从Python使用的一些良好且易于集成的单一文件数据库格式是SQLiteDBM变体之一。最后一个磁盘的行为就像字典(即您可以读取和写入键/值对)一样,但是将磁盘用作存储而不是1.2
GB的内存。