Update get grams

2022-06-30 12:11:59 +02:00 · 2022-06-30 12:11:59 +02:00 · c4cb117f5b
commit c4cb117f5b
parent 0ea3a9f18a
1 changed files with 6 additions and 1 deletions
--- a/getgrams.py
+++ b/getgrams.py
@ -42,8 +42,13 @@ if __name__ == "__main__":
    dataSorted = list(dataSorted.items())
    newDataSorted = []

+    if args.bigram:
+        reg = "[A-Za-z]{2}"
+    if args.trigram:
+        reg = "[A-Za-z]{3}"
+
    for entry in dataSorted:
-        if re.match("[A-Za-z]{2}", entry[0]):
+        if re.match(reg, entry[0]):
            newDataSorted.append((
                entry[0], entry[1]
            ))