From c4cb117f5befd0e674e1ea5d83a66a0e3140bda6 Mon Sep 17 00:00:00 2001 From: gbucchino Date: Thu, 30 Jun 2022 12:11:59 +0200 Subject: [PATCH] Update get grams --- getgrams.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/getgrams.py b/getgrams.py index 274ce03..a3c191e 100644 --- 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] ))