Python中yield的应用场景

Posted 2021-01-27 levylaw

使用yield可以构造一个生成器，可以在循环中避免前面的内容被后面的内容覆盖。例如：

import pandas as pd

dic_list = [{‘a‘:1,‘b‘:2,‘c‘:3},{‘a‘:4,‘b‘:5,‘c‘:6}]

def get_abc(dic_list):
    for item in dic_list:
        js = {}
        js[‘A‘] = item[‘a‘]
        js[‘B‘] = item[‘b‘]
        js[‘C‘] = item[‘c‘]
        yield(js)
        
pd.DataFrame(get_abc(dic_list))

output:
	A	B	C
0	1	2	3
1	4	5	6