根据空格和标点符号标记,保留标点符号

时间:2018-01-25 00:14:33

标签: python string tokenize

我正在寻找根据空格或标点符号进行标记或拆分的解决方案。结果中只能保留标点符号。它将用于识别语言(python,java,html,c ...)

输入string可以是:

class Foldermanagement():
def __init__(self):
    self.today = invoicemng.gettoday()
    ...

我期望的输出是列表/标记化,如下所述:

['class', 'Foldermanagement', '(', ')', ':', 'def', '_', '_', 'init', ... ,'self', '.', 'today', '=', ...]

欢迎任何解决方案,提前谢谢。

1 个答案:

答案 0 :(得分:3)

我认为这就是你要找的东西:

import string, re, itertools
text = """
class Foldermanagement():
def __init__(self):
    self.today = invoicemng.gettoday()
       """
separators = string.punctuation + string.whitespace
separators_re = "|".join(re.escape(x) for x in separators)
tokens = zip(re.split(separators_re, text), re.findall(separators_re, text))
flattened = itertools.chain.from_iterable(tokens)
cleaned = [x for x in flattened if x and not x.isspace()]
# ['class', 'Foldermanagement', '(', ')', ':', 'def', '_', '_',
#  'init', '_', '_', '(', 'self', ')', ':', 'self', '.', 'today', '=', 
#  'invoicemng', '.', 'gettoday', '(', ')']