在Python2.7.1中使用itertools、yield和iter()生成带有滑动窗口的字符串列表?

2024-06-01 00:44:21 发布

您现在位置:Python中文网/ 问答频道 /正文

我试图在Python中生成一个滑动窗口函数。我知道怎么做,但不是所有的功能。itertools、yield和iter()对我来说是全新的。

我想输入

a='abcdefg'
b=window(a,3)
print b
['abc','bcd','cde','def','efg']

我的工作方式是

def window(fseq, window_size=5):
    import itertools
    tentative=[]
    final=[]
    iteration=iter(fseq)
    value=tuple(itertools.islice(iteration,window_size))
    if len(value) == window_size:
        yield value
    for element in iteration:
        value = value[1:] + (element,)
        yield value

a='abcdefg'
result=window(a)
list1=[]
for k in result:
    list1.append(k)
list2=[]   
for j in list1:
    tentative=''.join(j)
    list2.append(tentative)
print list2

基本上,我所困惑的是如何在函数内部使用函数的最终值?

这是我的函数代码

def window(fseq, window_size=5):
    import itertools
    tentative=[]
    final=[]
    iteration=iter(fseq)
    value=tuple(itertools.islice(iteration,window_size))
    if len(value) == window_size:
        yield value
    for element in iteration:
        value = value[1:] + (element,)
        yield value
    for k in value:
        tentative.append(k)
    for j in tentative:
        tentative_string=''.join(j)
        final.append(tentative_string)
    return final



seq='abcdefg'
uence=window(seq)
print uence

我希望它返回加入的列表,但当我按下run时,它会显示“程序中有一个错误*'return',参数在生成器中”

我真的很困惑。


Tags: 函数inforsizevalueelementwindowfinal
3条回答

在一行代码中使用zip函数:

  [ "".join(j) for j in zip(*[fseq[i:] for i in range(window_size)])]

你是说你想这么做?以下内容:

a='abcdefg'
b = [a[i:i+3] for i in xrange(len(a)-2)]
print b
['abc', 'bcd', 'cde', 'def', 'efg']

你的发电机可能要短得多:

def window(fseq, window_size=5):
    for i in xrange(len(fseq) - window_size + 1):
        yield fseq[i:i+window_size]


for seq in window('abcdefghij', 3):
    print seq


abc
bcd
cde
def
efg
fgh
ghi
hij

相关问题 更多 >