正则-Strip函数

正则-Strip函数

@(正则表达式)[正则,strip]

strip()的正则表达式版本

写一个函数，它接受一个字符串，做的事情和strip()字符串方法一样。如果只传入了要去除的字符串，没有其他参数，那么就从该字符串首尾去除空白字符。否则，函数第二个参数指定的字符将从该字符串中去除。

import re


def strip(text, chars=None):
    """去除首尾的字符

    :type text: string
    :type chars: string
    :rtype: string
    """
    if chars is None:
        reg = re.compile('^ *| *$')
    else:
        reg = re.compile('^[' + chars + ']*|[' + chars + ']*$')
    return reg.sub('', text)


print(strip('   123456   '))  # 123456
print(strip('   123456'))  # 123456
print(strip('   123456'))  # 123456
print(strip('123456   654321'))  # 123456   654321
print(strip('123456   654321', '1'))  # 23456   65432
print(strip('123456   654321', '1234'))  # 56   65
print(strip('123456   654321', '124'))  # 3456   6543