Усреднение двух списков списков - PullRequest
Новая
       7

Усреднение двух списков списков

0 голосов
/ 25 ноября 2018

У меня есть два списка списков, и я хотел бы создать новый список списков, содержащий каждое среднее значение по списку (усреднение значений в каждом месте).например: если у меня есть следующие списки: 

 file1 = [[1, -1], [2, -2], [3, -3], [4, -4]]
 file2 = [[20, -40], [2, 0], [-50, 7], [30150, -200]]

, средний результат списка будет: (с использованием int для округления) 

avg = [[10, -20], [2, -1], [-23, 2], [15077, -102]]

, как только я это сделаю, еслисписки не имеют одинаковую длину, я должен добавить компоненты длинного списка, как есть.например: 

file1 = [[1, -1], [2, -2], [3, -3], [4, -4],[100, 100]
 file2 = [[20, -40], [2, 0], [-50, 7], [30150, -200]]

результат будет: 

[[10, -20], [2, -1], [-23, 2], [15077, -102],[100, 100]]

Я думал об использовании вложенных циклов, но это кажется слишком большим количеством операций.(поскольку эти списки являются лишь примерами, и я использую очень длинный список вместо этого ...) спасибо за помощь!

Ответы [ 4 ]

0 голосов
/ 25 ноября 2018

Я думаю, что это хорошее решение: 

def averageList(list1, list2):
    avg = []
    n = 0
    while n < max(len(list1), len(list2)):
        if n < min(len(list1), len(list2)):
            avg.append([int((list1[n][0]+list2[n][0])/2), int((list1[n][1]+list2[n][1])/2)])
        elif len(list1) > len(list2):
            avg.append(list1[n])
        elif len(list1) < len(list2):
            avg.append(list2[n])
        n = n+1
    return avg
0 голосов
/ 25 ноября 2018

Если вы хотите сделать это в одну строку, вы можете сделать что-то вроде: 

>>> [ [ int(sum(l2)/len(l2)) for l2 in zip(*l)] for l in zip(file1, file2) ]
[[10, -20], [2, -1], [-23, 2], [15077, -102]]

И мы можем делать уродливые вещи также с zip_longest: 

>>> [ [ int(sum([x for x in l2 if x is not None])/len([x for x in l2 if x is not None])) for l2 in zip_longest(*l)] for l in zip_longest(file1, file2, fillvalue=[]) ]
[[10, -20], [2, -1], [-23, 2], [15077, -102], [100, 100]]
0 голосов
/ 25 ноября 2018

Вы можете использовать zip_longest : 

from itertools import zip_longest

lst1 = [[1, -1], [2, -2], [3, -3], [4, -4], [100, 100]]
lst2 = [[20, -40], [2, 0], [-50, 7], [30150, -200]]


def avg(numbers):
    not_none = [n for n in numbers if n is not None]
    return sum(not_none) // len(not_none)


result = [[avg(e) for e in zip_longest(il, jl)] for il, jl in zip_longest(lst1, lst2, fillvalue=[])]

print(result)

Выход 

[[10, -21], [2, -1], [-24, 2], [15077, -102], [100, 100]]
0 голосов
/ 25 ноября 2018

Мы можем использовать zip() для объединения наших файлов, чтобы затем добавить среднее значение наших кортежей к avg. 

file1 = [[1, -1], [2, -2], [3, -3], [4, -4],[100, 100]]
file2 = [[20, -40], [2, 0], [-50, 7], [30150, -200]]
avg = []

for x1, x2 in zip(file1, file2):
    avg.append([(x1[0] + x2[0])//2, (x1[1] + x2[1])//2])

if len(avg) < len(file1):
    avg.extend(file1[len(avg):])
elif len(avg) < len(file2):
    avg.extend(file2[len(avg):])

print(avg) # -> [[10, -21], [2, -1], [-24, 2], [15077, -102], [100, 100]]
...