본문 바로가기

Series

Generator [Python] my_generator(), yield


# Python 3 : Generator


def my_generator():
yield 1
yield 2
yield 3


for i in my_generator():
print(i)


""" Output
1
2
3
"""



# Python 3 : Generator


def my_generator():
yield 1
yield 2
yield 3


gen = my_generator()
print(type(gen)) # <class 'generator'>

for i in gen:
print(i)

""" Output
<class 'generator'>
1
2
3
"""



# Python 3 : Generator


def my_generator():
yield 1
yield 2
yield 3


gen = my_generator()

print(next(gen)) # 1
print(next(gen)) # 2
print(next(gen)) # 3
print(next(gen)) # StopIteration


""" Output
1
2
3
StopIteration
"""