byte/discretemath-labs
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update networkbonus.py

Browse Source
This commit is contained in:
Daniel
2022-12-27 02:40:04 +02:00
committed by GitHub
parent f4b88a506f
commit 705a7fc7b2
1 changed files with 3 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
LabMD_3/networkbonus.py
View File

@@ -1,4 +1,3 @@
import json import json
import nltk import nltk
@@ -97,27 +96,18 @@ maxm = 0
for i in range(startingPoint, endingPoint): for i in range(startingPoint, endingPoint):
for x in wordsForCSV[i]: for x in wordsForCSV[i]:
graph_dict[x] = [] graph_dict[x] = []
filter_words = ["RT"] filter_words = ["RT", "of", "it", "not", "but", "for", "are", "in", "my", "to", "by", "doing", "so"]
file = open("data.csv", "w", encoding="utf-8") file = open("data.csv", "w", encoding="utf-8")
file.write("NODE,")
for i in range(startingPoint, endingPoint): for i in range(startingPoint, endingPoint):
for word in wordsForCSV[i]: for word in wordsForCSV[i]:
for x in wordsForCSV[i]: for x in wordsForCSV[i]:
if x is not word and x not in graph_dict[word] and x not in filter_words: if x is not word and x not in graph_dict[word] and x not in filter_words:
graph_dict[word].append(x) graph_dict[word].append(x)
for x in graph_dict: for x in graph_dict:
if len(graph_dict[x]) > maxm:
maxm = len(graph_dict[x])
for x in range(maxm):
file.write("EDGE" + str(x))
if x != maxm - 1:
file.write(",")
file.write("\n")
for x in graph_dict:
file.write(x)
file.write(",")
for z in range(len(graph_dict[x])): for z in range(len(graph_dict[x])):
file.write(x + ",")
file.write(graph_dict[x][z]) file.write(graph_dict[x][z])
if z != len(graph_dict[x]) - 1: if z != len(graph_dict[x]) - 1:
file.write(",") file.write(",")
file.write("\n")
file.write("\n") file.write("\n")