How to continue after the return command in python?

Question

Imagine we have a piece of code which cuts the large data into smaller data and do some process on it.

def node_cut(input_file):
    NODE_LENGTH = 500
    count_output = 0
    node_list=[]

    for line in input_file.readlines():
        if len(node_list) >= NODE_LENGTH :
            count_output += 1   
            return( node_list,count_output )
            node_list=[]  

        node,t=line.split(',')
        node_list.append(node) 

if __name__ =='__main__':

    input_data = open('all_nodes.txt','r')
    node_list, count_output = node_cut(input_data)
    some_process(node_list)

while node_cut return the first data list, the for loop stop going on for the rest of the large data. How I can make sure that it returns but still the loop continues?

NPE · Accepted Answer · 2011-10-25 10:05:14Z

7

Use yield:

def node_cut(input_file):
    NODE_LENGTH = 500
    count_output = 0
    node_list=[]

    for line in input_file.readlines():
        if len(node_list) >= NODE_LENGTH :
            count_output += 1   
            yield( node_list,count_output )
            node_list=[]  

        node,t=line.split(',')
        node_list.append(node) 

if __name__ =='__main__':
    with open('all_nodes.txt','r') as input_data:
      for node_list, count_output in node_cut(input_data):
        some_process(node_list)

answered Oct 25, 2011 at 10:05

NPE

503k114 gold badges970 silver badges1k bronze badges

Sign up to request clarification or add additional context in comments.

1 Comment

masti Over a year ago

Thanks for loop was a great advice!

Community · Accepted Answer · 2017-05-23 10:34:32Z

5

Use yield instead of return. See this question or this (somewhat old) article for how it works.

edited May 23, 2017 at 10:34

CommunityBot

11 silver badge

answered Oct 25, 2011 at 10:04

Fred Foo

365k80 gold badges765 silver badges852 bronze badges

Collectives™ on Stack Overflow

How to continue after the return command in python?

2 Answers 2

1 Comment

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

1 Comment

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related