正則-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