Writing each element of nested list into separate column of CSV

Question

I have 3 lists of n length that I want to combine into one list, then combine all the nth indexes together in a list, and produce the output with each element in a separate column in a csv file

list_1 = ["john", "peter", "steve", "mike", "paul"] 

list_2 = ["green", "red", "blue", "purple", "orange"] 

list_3 = [["dog", "cat"], "rabbit", "dog", "", ["cat", "mouse", "elephant"]

What should i do:

Answer 1

Easiest way to do it is to check if the last item is a list using the isinstance method, if so then extend it else use it as it:

list_1 = ["john", "peter", "steve", "mike", "paul"]

list_2 = ["green", "red", "blue", "purple", "orange"]

list_3 = [["dog", "cat"], "rabbit", "dog", "", ["cat", "mouse", "elephant"]]

 

combined_list = list(map(list, zip(list_1, list_2, list_3)))

for items in combined_list:

    if isinstance(items[-1], list):          #Check if last element is list.

        writer.writerow(items[:-1] + items[-1])

    else:

    writer.writerow(items)