Averages of elements in various lists

Question

I need to calculate the sum of several elements in different lists (from index 0).

lista1 = [154,12,998,147]
lista2 = [897,123,998,59877]
lista3 = [3,789,111,555,699]

That would be the simplistic way:

soma = lista1[0]+lista2[0]+lista3[0]

but I need to extract the values and apply the methods rolling().sum()

lista4 = pd.Series([154,897,3])
lista4.rolling(2).sum()

something like that !!

Answer 1

If you have a list of lists:

lista1 = [154,12,998,147]
lista2 = [897,123,998,59877]
lista3 = [3,789,111,555,699]

listas = [lista1, lista2, lista3]

You can create a list of the first elements by doing:

primeiros = [lista[0] for lista in listas]

Thus, primeiros will be the list [154, 897, 3].

As suggested in the comments, one possibility is to define a function that returns the values of a given list index:

def get_on_index(*listas, index=0):
    return [lista[index] for lista in listas]

So if you need all the values in index 2, just do:

valores = get_on_index(lista1, lista2, lista3, index=2)  # [998, 998, 555]