Modify function to accept tables as well (Python 3)

Question

I have designed the simple function that looks at an inputted list of numbers, identifies a minimum and maximum values, then substitutes both of them for amidpoint value between them, function is here:

def mainfunction(list_of_numbers):

    smallest_number = list_of_numbers[0]

    for a in list_of_numbers:

        if a < smallest_number:

            smallest_number = a

    largest_number = list_of_numbers[0]

    for b in list_of_numbers:

        if b > largest_number:

            largest_number = b

    midpoint = (smallest_number + largest_number)/2

    final_list = [x if (x != largest_number and x != smallest_number) else midpoint for x in list_of_numbers]

    return final_list

print(mainfunction([10, 7, 14, 3, -200, 8, 1, -12, 250]))

Unfortunately, I am not able to get the function to work on TABLES of numbers, is there any way to convert the table of numbers into the list?

Answer 1

You can use the itertools.chain

from itertools import chain

a = [[3,4], [15,16], [19,20]]

res = list(chain.from_iterable(a))

print(res)

Output:

[3, 4, 15, 16, 19, 20]

with list comprehension

res = [x for lst in a for x in lst]

 If you want to know more about the Data Science then do check out the following Data Science which will help you in understanding Data Science from scratch