get_same_mean

#!/usr/bin/env python3

"""
I am not sure if the implementation is correct, but you get back as result:
[[1,2,4,5]]
"""
from statistics import mean


def get_same_mean(data):
    results = []
    for i in data:
        new_data = data.copy()
        new_data.remove(i)
        if mean(data) == mean(new_data):
            if new_data in results:
                continue
            results.append(new_data)
    return results


data = list(range(1,6))
print('Data:')
print(data)
results = get_same_mean(data)
print('Results:')
print(results)


# output
# Data:
# [1, 2, 3, 4, 5]
# Results:
# [[1, 2, 4, 5]]