-4

我有一个很长的列表,我想替换字符串。我在下面做了一个简化的例子来说明我的问题。

my_list = ['a7_1_1', 'a7_2_1', 'a7_3_1','a7_1_2', 'a7_2_2', 'a7_3_2','a7_1_3', 'a7_2_3', 'a7_3_3']


Out[12]:
['a7_1_1',
 'a7_2_1',
 'a7_3_1',
 'a7_1_2',
 'a7_2_2',
 'a7_3_2',
 'a7_1_3',
 'a7_2_3',
 'a7_3_3'

我想用添加到前 3 个字符串的后缀替换字符串,因此最终列表应如下所示:

my_new_list = ['a7_1_1', 'a7_2_1', 'a7_3_1','a7_1_1.1', 'a7_2_1.1', 'a7_3_1.1','a7_1_1.2', 'a7_2_1.2', 'a7_3_1.2']


Out[15]:
['a7_1_1',
 'a7_2_1',
 'a7_3_1',
 'a7_1_1.1',
 'a7_2_1.1',
 'a7_3_1.1',
 'a7_1_1.2',
 'a7_2_1.2',
 'a7_3_1.2']

是否有捷径可寻?

4

2 回答 2

2

使用 itertools.cycle() 函数

import itertools as it           #1

def cycle_first_n(lst, n):
    """ cycles through first n elements of the list """
    c = it.cycle(lst[:n])        #2
    for idx in xrange(len(lst)): #3
        sfx = idx / n
        yield c.next() + ('.' + str(sfx) if sfx > 0 else '')  #4
  1. itertools 是一个用于创建迭代器以实现高效循环的库
  2. 创建一个迭代器以循环遍历列表的 n 个元素的切片
  3. 使用 xrange 而不是 range 来避免在内存中创建可能很长的列表(请参阅问题)
  4. yield 意味着我们正在创建一个生成器。再次避免在内存中创建长列表

如何使用该功能

lst = ['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h']
for o in cycle_first_n(lst, 3):
    print o,

输出

a b c a.1 b.1 c.1 a.2 b.2
于 2013-09-10T06:13:54.323 回答
0

我不是很清楚你的意思。检查您是否要这样做:

>>> my_list = ['a7_1_1', 'a7_2_1', 'a7_3_1','a7_1_2', 'a7_2_2', 'a7_3_2','a7_1_3', 'a7_2_3', 'a7_3_3']
>>> my_new_list = sum([[x, x+'.1', x+'.2'] for x in my_list[:3]], [])
>>> print(my_new_list)
['a7_1_1', 'a7_1_1.1', 'a7_1_1.2', 'a7_2_1', 'a7_2_1.1', 'a7_2_1.2', 'a7_3_1', 'a7_3_1.1', 'a7_3_1.2']
于 2013-09-10T05:37:47.503 回答