Update get grams
This commit is contained in:
parent
0ea3a9f18a
commit
c4cb117f5b
@ -42,8 +42,13 @@ if __name__ == "__main__":
|
|||||||
dataSorted = list(dataSorted.items())
|
dataSorted = list(dataSorted.items())
|
||||||
newDataSorted = []
|
newDataSorted = []
|
||||||
|
|
||||||
|
if args.bigram:
|
||||||
|
reg = "[A-Za-z]{2}"
|
||||||
|
if args.trigram:
|
||||||
|
reg = "[A-Za-z]{3}"
|
||||||
|
|
||||||
for entry in dataSorted:
|
for entry in dataSorted:
|
||||||
if re.match("[A-Za-z]{2}", entry[0]):
|
if re.match(reg, entry[0]):
|
||||||
newDataSorted.append((
|
newDataSorted.append((
|
||||||
entry[0], entry[1]
|
entry[0], entry[1]
|
||||||
))
|
))
|
||||||
|
Loading…
Reference in New Issue
Block a user