Matrix+TXT 2

"""
Пример текстового документа:
25 17
31 58
73 42
"""
# Создаём двумерный список чисел, которые берём из текстового документа
matrix = []
with open("example.txt") as f:
    for line in f:
        matrix.append([int(x) for x in line.split()])


# Вывод в консоль сгенерированной матрицы
m = len(matrix)
k = len(matrix[0])

print("\n\nДвумерный массив: ")
for i in range(m):
    for j in range(k):
        print(matrix[i][j], end = ' ')
    print()


# Нахождение максимального значения в каждом столбце и последующее занесение этих чисел в список maximum
maximum = list(map(max, zip(*matrix)))

# Нахождение среднего арифметического каждого столбца и последующее занесение этих чисел в список mean
mean = list(map(lambda x: sum(x)/len(x), zip(*matrix)))


# Вывод в консоль максимальных значений каждого столбца
print("Максимальные числа каждого столбца:")
for element in maximum:
    print(element, end=" ")
print()
# Вывод в консоль среднего арифметического каждого столбца
print("Среднее арифметическое каждого столбца:")
for element in mean:
    print(element, end=" ")
print()


# Запись максимальных значений под соответствующим столбцом
# Запись Больший/Меньший под столбцами, среднее арифметическое которых соответственно больше/меньше другого
with open ("example.txt", "a") as f:
    f.write("\n\n")
    for element in maximum:
        f.write(str(element)+" ")
    f.write("\n")
    for element in mean:
        if element == max(mean):
            f.write("Больший ")
        else:
            f.write("Меньший ")