Python:从循环中返回值而不会中断

时间:2012-10-29 19:38:18

标签: python return return-value yield

G'day,我有一份按地点分组的个人名单。我想生成一个新变量,根据每个人的位置为每个人提供一个数字。我想要的数据是:

place       individual
here        1
here        2
here        3
there       1
there       2
somewhere   1 
somewhere   2

我写了这个:

    nest="ddd", "ddd", "fff", "fff", "fff", "fff", "qqq", "qqq"

    def individual(x):
        i = 0
        j = 1
        while i < len(x):
            if x[i] == x[i-1]:
                print(j+1)
                i = i + 1
                j = j + 1
            else:
                print(1)
                i = i + 1
                j = 1

    individual(nest)

这会打印出我想要的值,但是,当我将返回放在那里时,它会突破循环并仅返回第一个值。我想知道如何返回这些值,以便我可以将它们作为新列添加到我的数据中?

我读到了收益率?但不确定它是否合适。谢谢你的帮助!

干杯, 亚当

1 个答案:

答案 0 :(得分:5)

print(...)替换为yield ...。然后你会有一个生成器,它会给你一个可迭代的。然后,您可以通过迭代结果将其转换为其他适当的数据结构。例如,要从生成器构造列表,您可以执行以下操作:

list(individual(nest))  #this is prefered

在这种情况下,迭代是隐含的......

或(在这种背景下更圆整但可能更具信息性):

[ x for x in individual(nest) ]  #This is just for demonstration.