Python中的反轉字符串問題

按單詞反轉字符串是一道很常見的面試題。在Python中實現起來非常簡單。

def reverse_string_by_word(s):
    lst = s.split()  # split by blank space by default
    return ' '.join(lst[::-1])

s = 'Power of Love'
print reverse_string_by_word(s)
# Love of Power

s = 'Hello    World!'
print reverse_string_by_word(s)
# World! Hello

上面的實現其實已經能滿足大多數情況，但是並不完美。比如第二個字符串中的感嘆號並沒有被翻轉，而且原字符串中的空格數量也沒有保留。（在上面的例子裏其實Hello和World之間不止一個空格）

我們期望的結果應該是這樣子的。

print reverse_string_by_word(s)
# Expected: !World  Hello

要改進上面的方案還不把問題複雜化，推薦使用re模塊。你可以查閱re.split() 的官方文檔。我們看一下具體例子。

>>> import re
>>> s = 'Hello  World!'

>>> re.split(r'\s+', s)    # will discard blank spaces
['Hello', 'World!']

>>> re.split(r'(\s+)', s)  # will keep spaces as a group
['Hello', '  ', 'World!']

>>> s = '< Welcome to EF.COM! >'

>>> re.split(r'\s+', s)  # split by spaces
['<', 'Welcome', 'to', 'EF.COM!', '>']

>>> re.split(r'(\w+)', s)  # exactly split by word
['< ', 'Welcome', ' ', 'to', ' ', 'EF', '.', 'COM', '! >']

>>> re.split(r'(\s+|\w+)', s)  # split by space and word
['<', ' ', '', 'Welcome', '', ' ', '', 'to', '', ' ', '', 'EF', '.', 'COM', '!', ' ', '>']

>>> ''.join(re.split(r'(\s+|\w+)', s)[::-1])
'> !COM.EF to Welcome <'

>>> ''.join(re.split(r'(\s+)', s)[::-1])
'> EF.COM! to Welcome <'

>>> ''.join(re.split(r'(\w+)', s)[::-1])
'! >COM.EF to Welcome< '

如果你覺得用切片將序列倒序可讀性不高，那麼其實也可以這樣寫。

>>> ''.join(reversed(re.split(r'(\s+|\w+)', s)))
'> !COM.EF to Welcome <'

一句話搞定，so easy!

關於作者：Python技術愛好者，目前從事測試開發相關工作，轉載請註明原文出處。

歡迎關注我的博客 http://betacat.online，你可以到我的公衆號中去當吃瓜羣衆。