Passing multiple lists through a function

Question

I would like to use 3 lists as arguments of a function. Maybe my entire logic is wrong on this, but what I'm trying to do is use two lists to create the iterators to parse the third list. The order is important, so the first element of the first and second list would give the first and last index etc.

My code gives only the first sublist, when of course I would like to have all of them.

A=[1, 3, 4, 6]
B=[3, 8, 5, 8]
C=['A', 'B','C', 'D', 'E', 'A', 'B', 'A', 'D']

def newlists(entirelist, initiating, terminating):
    for initindex in initiating:
        for termindex in terminating:

        parsed_list = entirelist[initindex:termindex]

        return parsed_list



all_sublists = newlists(C, A, B)

I need the following output or something similar to obtain the sublists:

all_sublists = [['B','C'],['D','E', 'A', 'B', 'A'],['E'],['B','A']]

I think you will first need to define parsed_list = [] at the beginning of your function and then, instead of parsed_list = entirelist[initindex:termindex] use this parsed_list.append(entirelist[initindex:termindex]) — NutCracker
– NutCracker, Commented Nov 1, 2018 at 9:23

zipa · Accepted Answer · 2018-11-01 09:22:33Z

1

You can use comprehension to get what you want:

A=[1, 3, 4, 6]
B=[3, 8, 5, 8]
C=['A', 'B','C', 'D', 'E', 'A', 'B', 'A', 'D']

def newlists(entirelist, initiating, terminating):
    return [entirelist[i:j] for i, j in zip(initiating, terminating)]

print(newlists(C, A, B))
#[['B', 'C'], ['D', 'E', 'A', 'B', 'A'], ['E'], ['B', 'A']]

answered Nov 1, 2018 at 9:22

zipa

28k6 gold badges45 silver badges62 bronze badges

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

Comments

englealuze · Accepted Answer · 2018-11-07 12:55:41Z

0

A solution without list comprehension and fancy zip function:

def newlists(entirelist, initiating, terminating):
    return map(lambda i, j: entirelist[i:j], initiating, terminating)

print(list(newlists(C,A,B)))
# => [['B', 'C'], ['D', 'E', 'A', 'B', 'A'], ['E'], ['B', 'A']]

answered Nov 7, 2018 at 12:55

englealuze

1,7631 gold badge14 silver badges24 bronze badges

Collectives™ on Stack Overflow

Passing multiple lists through a function

2 Answers 2

Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related