Как посчитать сумму чисел в матрице?

Question

[Muriam]

#import csv

# чтение матрицы из файла
with open('in.txt') as f:
    matrix = [list(map(int, row.split())) for row in f.readlines()]

print(matrix)
    
s = sum(matrix)
print(s)


# сохранение матрицы в другой файл    
with open('output.txt', 'w') as f:
    for row in matrix:
        f.write(' '.join(map(str, row)) + '\n')

например, такой код работает
chars = [1,2,3,4,5]
s = sum(chars) # 15
print(s)
а для двумерных списков, нет

Answer 1

[kova1ev]

можно так

import functools

matrix = [[]]

result = functools.reduce(lambda a,x: a + sum(x), matrix, 0)
print(result)

но это если матрица как двумерный массив (даже если пустой), если матрица будет просто списком, будет ошибка. Как сделать более грамотно, может сейчас опытные питонисты подскажут.

Comments

[longclaps]

print(sum(map(sum, matrix)))

[Muriam]

longclaps, это верно! и компактно)

[Muriam]

longclaps,
как убрать лишние скобки в выводе?
как перенести каждую строку матрицы с новой строки в выводе?

у меня немного не красиво:

with open('in.txt') as f:
    matrix = [list(map(int, row.split())) for row in f.readlines()]

print(matrix)

print(sum(map(sum, matrix)))

[longclaps]

у меня немного не красиво

так и должно быть

как перенести каждую строку матрицы с новой строки в выводе?

from pprint import pprint

pprint([[1, 0, 0], [0, 1, 0], [0, 0, 1]], width=20)

[Muriam]

longclaps,

вот такой цикл удовлетворяет моему запросу красоты:

#print(*matrix)      это лишнее, гораздо красивее отображение, см цикл ниже
for i in matrix:
    for j in i:
        print(j,end=' ')
    print()

при этом матрица отобразится, точно как в файле

Answer 2

[Denis_Yav]

with open('in.txt') as f:
matrix = [list(map(int, row.split())) for row in f.readlines()]

print(matrix)
for m in matrix:
s = sum(m)
print(s)

Comments

[kova1ev]

s = 0
for m in matrix:
    s += sum(m)

[Muriam]

Denis_Yav ваше суммирование дает результат 4,
не плюсуется последняя третья строка матрицы
kova1ev ваше суммирование дает результат 8 - это верно!

вывод марицы на экран
print(matrix)
дает такой результат:
как сделать, чтоб матица была записана не в одну строку, а чтоб переносилась после каждых нескольких элементов на новую строку, как это записано в файле?
и как сделать. чтоб не высвечивались скобки у выведенной матрицы?

код:

with open('in.txt') as f:
    matrix = [list(map(int, row.split())) for row in f.readlines()]

s = 0
for m in matrix:
    s += sum(m)
print(s)

print(matrix)

Answer 3

[Вадим Шаталов]

Не изобретайте велосипеды
https://docs.scipy.org/doc/numpy/reference/generat...

Comments

[Denis_Yav]

спасибо. мне хоть и не нужно было сумму в матрице, но ссылка полезна