如何将列表元素拆分为两个？

Question

如果我有数据存储在列表中，例如

images = ['pdf-one','gif-two','jpg-three'] 

如何在连字符中将这些元素拆分为多个元素 - 而不是子列表。即

images = ['pdf','-one','gif','-two','jpg','-three'] 

不

images = [['pdf','-one'],['gif','-two'],['jpg','-three']] 

Answer 1

在用正则表达式这种情况下分裂使得对最可读的代码：

import re 

hyphensplit = re.compile('(-[a-z]+)').split 
images = [part for img in images for part in hyphensplit(img) if part] 

演示：

>>> import re 
>>> hyphensplit = re.compile('(-[a-z]+)').split 
>>> images = ['pdf-one','gif-two','jpg-three'] 
>>> [part for img in images for part in hyphensplit(img) if part] 
['pdf', '-one', 'gif', '-two', 'jpg', '-three'] 

Answer 2

您可以使用str.partition为这个：

>>> from itertools import chain 
>>> images = ['pdf-one', 'gif-two', 'jpg-three'] 
>>> list(chain.from_iterable([[a, b+c] for a, b, c 
              in (x.partition('-') for x in images)])) 
['pdf', '-one', 'gif', '-two', 'jpg', '-three'] 

使用发电机功能进行更多可读溶液：

def my_split(seq): 
    for item in seq: 
     a, b, c = item.partition('-') 
     yield a 
     yield b+c 

>>> list(my_split(images)) 
['pdf', '-one', 'gif', '-two', 'jpg', '-three']