Как найти сумму элементов двумерного массива python

На уроке рассматриваются алгоритмы работы с двумерными массивами в Python: создание матрицы, инициализация элементов, вывод, обработка элементов матрицы

Создание, вывод и ввод матрицы в Питоне

Для работы с матрицами в Python также используются списки. Каждый элемент списка-матрицы содержит вложенный список.

• Таким образом, получается структура из вложенных списков, количество которых определяет количество строк матрицы, а число элементов внутри каждого вложенного списка указывает на количество столбцов в исходной матрице.

Рассмотрим пример матрицы размера 4 х 3:

matrix = [[-1, 0, 1], 
    [-1, 0, 1], 
    [0, 1, -1],
    [1, 1, -1]]

Данный оператор можно записать в одну строку:

matrix = [[-1, 0, 1], [-1, 0, 1], [0, 1, -1], [1, 1, -1]]

• Вывод матрицы можно осуществить одним оператором, но такой простой способ не позволяет выполнять какой-то предварительной обработки элементов:

Результат:
 

1. способ:

1 2 3 4 5

def printMatrix ( matrix ): for i in range ( len(matrix) ): for j in range ( len(matrix[i]) ): print ( "{:4d}".format(matrix[i][j]), end = "" ) print ()

В примере i – это номер строки, а j – номер столбца;
len(matrix) – число строк в матрице.

2. способ:

1 2 3 4 5

def printMatrix ( matrix ): for row in matrix: for x in row: print ( "{:4d}".format(x), end = "" ) print ()

Внешний цикл проходит по строкам матрицы (row), а внутренний цикл проходит по элементам каждой строки (x).

1
2
3
4

from random import randint
n, m = 3, 3
a = [[randint(1, 10) for j in range(m)] for i in range(n)]
print(a)

Обработка элементов двумерного массива

Нумерация элементов двумерного массива, как и элементов одномерного массива, начинается с нуля.
Т.е. matrix[2][3] — это элемент третьей строки четвертого столбца.

Пример обработки элементов матрицы:
Найти произведение элементов двумерного массива.

✍ Решение:
 

1
2
3
4
5

p = 1 
for i in range(N): 
    for j in range(M): 
       p *= matrix[i][j] 
print (p)

Пример:
Найти сумму элементов двумерного массива.

✍ Решение:
 

1
2
3
4

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

Задание Python 8_1:
Написать программу поиска минимального и максимального элементов матрицы и их индексов.

Задание Python 8_2:
Написать программу, выводящую на экран строку матрицы, сумма элементов которой максимальна.

for i in range(N): 
   # работаем с matrix[i][i]

for i in range(N): 
   # работаем с matrix[i][N-1-i]

Пример:Переставить 2-й и 4-й столбцы матрицы. Использовать два способа.

✍ Решение:
 

Improve Article

Save Article

Like Article

Given a 2-D matrix, we need to find sum of all elements present in matrix ? Examples:

Input  : arr = [[1, 2, 3], 
                [4, 5, 6], 
                [2, 1, 2]]
Output : Sum = 26

This problem can be solved easily using two for loops by iterating whole matrix but we can solve this problem quickly in python using map() function. 

What does map() do? 
The map() function applies a given function to each item of an iterable(list, tuple etc.) and returns a list of the results. For example see given below example : 

<map object at 0x7fdf95d2a6d8>
[1, 4, 9, 16]

This article is contributed by Shashank Mishra (Gullu). If you like GeeksforGeeks and would like to contribute, you can also write an article using write.geeksforgeeks.org or mail your article to review-team@geeksforgeeks.org. See your article appearing on the GeeksforGeeks main page and help other Geeks.

Last Updated :
21 Jul, 2022

Like Article

Save Article

• Матрицы в Python
• NumPy массивы в Python
• Как создать массив NumPy?
  • Массив целых чисел, чисел с плавающей точкой и составных чисел
  • Массив нулей и единиц
  • Использование arange() и shape()
• Операции с матрицами
  • Сложение двух матриц или сумма элементов массива Python
  • Умножение двух матриц Python
  • Транспонирование матрицы питон
• Доступ к элементам матрицы, строкам и столбца
  • Доступ к элементам матрицы
  • Доступ к строкам матрицы
  • Доступ к столбцам матрицы
• Разделение матрицы

A = [[1, 4, 5], 
    [-5, 8, 9]]

A = [[1, 4, 5, 12], 
    [-5, 8, 9, 0],
    [-6, 7, 11, 19]]

print("A =", A) 
print("A[1] =", A[1])      # вторая строка
print("A[1][2] =", A[1][2])   # третий элемент второй строки
print("A[0][-1] =", A[0][-1])   # последний элемент первой строки

column = [];        # пустой список
for row in A:
  column.append(row[2])   

print("3rd column =", column)

A = [[1, 4, 5, 12], [-5, 8, 9, 0], [-6, 7, 11, 19]]
A [1] = [-5, 8, 9, 0]
A [1] [2] = 9
A [0] [- 1] = 12
3-й столбец = [5, 9, 11]

import numpy as np
a = np.array([1, 2, 3])
print(a)               # Вывод: [1, 2, 3]
print(type(a))         # Вывод: <class 'numpy.ndarray'>

import numpy as np

A = np.array([[1, 2, 3], [3, 4, 5]])
print(A)

A = np.array([[1.1, 2, 3], [3, 4, 5]]) # Массив чисел с плавающей запятой
print(A)

A = np.array([[1, 2, 3], [3, 4, 5]], dtype = complex) # Массив составных чисел
print(A)

[[1 2 3]
 [3 4 5]]

[[1.1 2. 3.]
 [3. 4. 5.]]

[[1. + 0.j 2. + 0.j 3. + 0.j]
 [3. + 0.j 4. + 0.j 5. + 0.j]]

import numpy as np

zeors_array = np.zeros( (2, 3) )
print(zeors_array)

'''
 Вывод:
 [[0. 0. 0.]
  [0. 0. 0.]]
'''

ones_array = np.ones( (1, 5), dtype=np.int32 ) // указание dtype
print(ones_array)      # Вывод: [[1 1 1 1 1]]

import numpy as np

A = np.arange(4)
print('A =', A)

B = np.arange(12).reshape(2, 6)
print('B =', B)

''' 
Вывод:
A = [0 1 2 3]
B = [[ 0  1  2  3  4  5]
 [ 6  7  8  9 10 11]]
'''

import numpy as np

A = np.array([[2, 4], [5, -6]])
B = np.array([[9, -3], [3, 6]])
C = A + B      # сложение соответствующих элементов
print(C)

''' 
Вывод:
[[11  1]
 [ 8  0]]
 '''

import numpy as np

A = np.array([[3, 6, 7], [5, -3, 0]])
B = np.array([[1, 1], [2, 1], [3, -3]])
C = a.dot(B)
print(C)

''' 
Вывод:
[[ 36 -12]
 [ -1   2]]
'''

import numpy as np

A = np.array([[1, 1], [2, 1], [3, -3]])
print(A.transpose())

''' 
Вывод:
[[ 1  2  3]
 [ 1  1 -3]]
'''

import numpy as np
A = np.array([2, 4, 6, 8, 10])

print("A[0] =", A[0])     # Первый элемент     
print("A[2] =", A[2])     # Третий элемент     
print("A[-1] =", A[-1])   # Последний элемент     

A [0] = 2
A [2] = 6
A [-1] = 10

import numpy as np

A = np.array([[1, 4, 5, 12],
    [-5, 8, 9, 0],
    [-6, 7, 11, 19]])

#  Первый элемент первой строки
print("A[0][0] =", A[0][0])  

# Третий элемент второй строки
print("A[1][2] =", A[1][2])

# Последний элемент последней строки
print("A[-1][-1] =", A[-1][-1])     

A [0] [0] = 1
A [1] [2] = 9
A [-1] [- 1] = 19

import numpy as np

A = np.array([[1, 4, 5, 12], 
    [-5, 8, 9, 0],
    [-6, 7, 11, 19]])

print("A[0] =", A[0]) # Первая строка
print("A[2] =", A[2]) # Третья строка
print("A[-1] =", A[-1]) # Последняя строка (третья строка в данном случае)

A [0] = [1, 4, 5, 12]
A [2] = [-6, 7, 11, 19]
A [-1] = [-6, 7, 11, 19]

import numpy as np

A = np.array([[1, 4, 5, 12], 
    [-5, 8, 9, 0],
    [-6, 7, 11, 19]])

print("A[:,0] =",A[:,0]) # Первый столбец
print("A[:,3] =", A[:,3]) # Четвертый столбец
print("A[:,-1] =", A[:,-1]) # Последний столбец (четвертый столбец в данном случае)

A [:, 0] = [1 -5 -6]
A [:, 3] = [12 0 19]
A [:, - 1] = [12 0 19]

import numpy as np
letters = np.array([1, 3, 5, 7, 9, 7, 5])

# с 3-го по 5-ый элементы
print(letters[2:5])        # Вывод: [5, 7, 9]

# с 1-го по 4-ый элементы
print(letters[:-5])        # Вывод: [1, 3]   

# с 6-го до последнего элемента
print(letters[5:])         # Вывод:[7, 5]

# с 1-го до последнего элемента
print(letters[:])          # Вывод:[1, 3, 5, 7, 9, 7, 5]

# список в обратном порядке
print(letters[::-1])          # Вывод:[5, 7, 9, 7, 5, 3, 1]

import numpy as np

A = np.array([[1, 4, 5, 12, 14], 
    [-5, 8, 9, 0, 17],
    [-6, 7, 11, 19, 21]])

print(A[:2, :4])  # две строки, четыре столбца

''' Вывод:
[[ 1  4  5 12]
 [-5  8  9  0]]
'''

print(A[:1,])  # первая строка, все столбцы

''' Вывод:
[[ 1  4  5 12 14]]
'''

print(A[:,2])  # все строки, второй столбец

''' Вывод:
[ 5  9 11]
'''

print(A[:, 2:5])  # все строки, с третьего по пятый столбец
''' Вывод:
[[ 5 12 14]
 [ 9  0 17]
 [11 19 21]]
'''