-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathmk.py
47 lines (38 loc) · 1.28 KB
/
mk.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
from urllib.request import urlopen
from random import randint
def wordListSum(wordList):
sum = 0
for word, value in wordList.items():
sum += value
return sum
def retrieveRandomWord(wordList):
randIndex = randint(1, wordListSum(wordList))
for word, value in wordList.items():
randIndex -= value
if randIndex <= 0:
return word
def buildWordDict(text):
text = text.replace('\n', ' ')
text = text.replace('\"', '')
pubctuation = [',', '.', ';', ':']
for symbol in pubctuation:
text = text.replace(symbol, ' '+symbol+' ')
words = text.split(' ')
words = [word for word in words if word != '']
wordDict = {}
for i in range(1, len(words)):
if words[i-1] not in wordDict:
wordDict[words[i-1]] = {}
if words[i] not in wordDict[words[i-1]]:
wordDict[words[i-1]][words[i]] = 0
wordDict[words[i-1]][words[i]] = wordDict[words[i-1]][words[i]]+1
return wordDict
text = str(urlopen("http://pythonscraping.com/files/inaugurationSpeech.txt").read(), 'utf-8')
wordDict = buildWordDict(text)
length = 100
chain = ''
currentWord = 'I'
for i in range(0, length):
chain += currentWord+' '
currentWord = retrieveRandomWord(wordDict[currentWord])
print(chain)