Печатает пары элементов в списке, которые имеют то же среднее значение, что и среднее значение списка

Question

Я написал простой код для вычисления среднего значения списка.

import statistics

x = [5, 6, 7, 0, 3, 1, 7]
print(round(statistics.mean(x)))

>>>> 4

Как заставить его печатать пары с одинаковым средним значением?Например, [1, 7] имеет то же среднее значение, что и 4.

Answer 1

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

from itertools import combinations
import numpy as np

def close(p, q, eps): # Use to compare if 2 float values are closer than epsilon
    return np.abs(p - q) <= eps

l = [5, 6, 7, 0, 3, 1, 7]
my_mean = np.round(np.mean(l)) # 4.0
a = [(x, y) for (x, y) in combinations(l, 2) if close(np.mean([x,y]), my_mean, 0.1)]

print(a) # [(5, 3), (7, 1), (1, 7)]

Answer 2

Вы можете попробовать это.

import random
while True:
    lst = random.sample(range(15), random.randint(2,10)) 
    if sum(lst)/len(lst) == 4:
        print(lst) 

Примечание: этот метод может создавать дубликаты списков.

Answer 3

>>> from itertools import permutations, combinations
>>> l = [5, 6, 7, 0, 3, 1, 7] 
# compute average
>>> avg = sum(l)//len(l) 
# generate all possible combinations
>>> [i for i in combinations(l, 2)] 
[(5, 6), (5, 7), (5, 0), (5, 3), (5, 1), (5, 7), (6, 7), (6, 0), (6, 3), (6, 1), (6, 7), (7, 0), (7, 3), (7, 1), (7, 7), (0, 3), (0, 1), (0, 7), (3, 1), (3, 7), (1, 7)]
>>> [(a+b)//2 for a,b in combinations(l, 2)] 
[5, 6, 2, 4, 3, 6, 6, 3, 4, 3, 6, 3, 5, 4, 7, 1, 0, 3, 2, 5, 4]
# only filter those which average to the mean of the whole list
>>> [(a,b) for a,b in combinations(l, 2) if (a+b)//2==avg]  
[(5, 3), (6, 3), (7, 1), (1, 7)]

Answer 4

Парная итерация

for a, b in itertools.product(x, x):
    if (a + b) / 2 == average:
        print(a, b)