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

На уроке рассматриваются алгоритмы работы с двумерными массивами в 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

Массив — это набор линейных структур данных, которые содержат все элементы одного типа данных в непрерывном пространстве памяти. Это похоже на контейнер, содержащий определенное количество элементов с одинаковым типом данных. Индекс массива начинается с 0, и поэтому программист может легко получить положение каждого элемента и выполнить различные операции с массивом. В этом разделе мы узнаем о 2D (двумерных) массивах в Python.

Двумерный массив (2D-массив)

2D-массив — это массив из массивов, который может быть представлен в матричной форме, такой как строки и столбцы. В этом массиве положение элементов данных определяется двумя индексами вместо одного индекса.

Синтаксис:

  Array_name = [rows] [columns] # объявление 2D-массива
 Arr-name = [[m1, m2, m3,…. mn], [n1, n2, n3,… .. nn]]

Где m — строка, а n — столбец таблицы.

Доступ к двумерному массиву

В Python мы можем получить доступ к элементам двумерного массива с помощью двух индексов. Первый индекс относится к индексации списка, а второй индекс относится к положению элементов. Если мы определяем только один индекс с именем массива, он возвращает все элементы 2-мерного массива, хранящиеся в массиве.

Давайте создадим простую программу для понимания 2D (двумерных) массивов на Python.

2dSimple.py

 
Student_dt = [ [72, 85, 87, 90, 69], [80, 87, 65, 89, 85], [96, 91, 70, 78, 97], [90, 93, 91, 90, 94], [57, 89, 82, 69, 60] ] 
#print(student_dt[]) 
print(Student_dt[1]) # print all elements of index 1 
print(Student_dt[0]) # print all elements of index 0 
print(Student_dt[2]) # print all elements of index 2 
print(Student_dt[3][4]) # it defines the 3rd index and 4 position of the data element. 

Выход:

В приведенном выше примере мы передали 1, 0 и 2 в качестве параметров в 2D-массив, который печатает всю строку определенного индекса. И мы также передали student_dt [3] [4], который представляет 3-й индекс и 4-ю позицию двумерного массива элементов для печати конкретного элемента.

Перемещение элемента в 2D-массиве

Program.py

 
# write a program to traverse every element of the two-dimensional array in Python. 
Student_dt = [ [72, 85, 87, 90, 69], [80, 87, 65, 89, 85], [96, 91, 70, 78, 97], [90, 93, 91, 90, 94], [57, 89, 82, 69, 60] ] 
# Use for loop to print the entire elements of the two dimensional array. 
for x in Student_dt:  # outer loop 
    for i in x:  # inner loop 
        print(i, end = " ") # print the elements 
    print() 

Выход:

Вставка элементов в двумерный массив

Мы можем вставлять элементы в двумерный массив, используя функцию insert(), которая указывает номер индекса элемента и местоположение для вставки.

Insert.py

 
# Write a program to insert the element into the 2D(two dimensional) array of Python. 
from array import * # import all package related to the array. 
arr1 = [[1, 2, 3, 4], [8, 9, 10, 12]]  # initialize the array elements. 
print("Before inserting the array elements: ") 
print(arr1) # print the arr1 elements. 
# Use the insert() function to insert the element that contains two parameters. 
arr1.insert(1, [5, 6, 7, 8])  # first parameter defines the index no., and second parameter defines the elements 
print("After inserting the array elements ") 
for i in arr1: # Outer loop 
    for j in i: # inner loop 
        print(j, end = " ") # print inserted elements. 
    print()     

Выход:

Обновление элементов в массиве

В 2D-массиве существующее значение массива может быть обновлено новым значением. В этом методе мы можем изменить конкретное значение, а также весь индекс массива.

Создадим программу для обновления существующего значения 2D-массива в Python.

Update.py

 
from array import * # import all package related to the array. 
arr1 = [[1, 2, 3, 4], [8, 9, 10, 12]]  # initialize the array elements. 
print("Before inserting the array elements: ") 
print(arr1) # print the arr1 elements. 
 
 
arr1[0] = [2, 2, 3, 3] # update the value of the index 0 
arr1[1][2] = 99 # define the index [1] and position [2] of the array element to update the value. 
print("After inserting the array elements ") 
for i in arr1: # Outer loop 
    for j in i: # inner loop 
        print(j, end = " ") # print inserted elements. 
    print()  

Выход:

Удаление значения из 2D-массива в Python

В двумерном массиве мы можем удалить конкретный элемент или весь индекс массива с помощью функции del() в Python. Давайте разберемся с примером удаления элемента.

Delete.py

 
from array import * # import all package related to the array. 
arr1 = [[1, 2, 3, 4], [8, 9, 10, 12]]  # initialize the array elements. 
print("Before Deleting the array elements: ") 
print(arr1) # print the arr1 elements. 
 
del(arr1[0][2]) # delete the particular element of the array. 
del(arr1[1]) # delete the index 1 of the 2-D array. 
 
print("After Deleting the array elements ") 
for i in arr1: # Outer loop 
    for j in i: # inner loop 
        print(j, end = " ") # print inserted elements. 
    print()     

Выход:

Размер двумерного массива

Функция len() используется для получения длины двумерного массива. Другими словами, мы можем сказать, что функция len() определяет общий индекс, доступный в двумерных массивах.

Давайте разберемся с функцией len(), чтобы получить размер двумерного массива в Python.

Size.py

 
array_size = [[1, 3, 2],[2,5,7,9], [2,4,5,6]] # It has 3 index 
print("The size of two dimensional array is : ") 
print(len(array_size)) # it returns 3  
 
array_def = [[1, 3, 2], [2, 4, 5, 6]] # It has 2 index 
print("The size of two dimensional array is : ") 
print(len(array_def)) # it returns 2 

Выход:

Напишем программу для вывода суммы двумерных массивов на Python.

Matrix.py

 
def two_d_matrix(m, n): # define the function 
    Outp = []  # initially output matrix is empty 
    for i in range(m): # iterate to the end of rows 
        row = [] 
        for j in range(n): # j iterate to the end of column 
            num = int(input(f "Enter the matrix [{0}][{j}]")) 
            row.append(num) # add the user element to the end of the row 
        Outp.append(row) # append the row to the output matrix 
    return Outp     
 
def sum(A, B): # define sum() function to add the matrix. 
    output = [] # initially, it is empty. 
    print("Sum of the matrix is :") 
    for i in range(len(A)): # no. of rows 
        row = [] 
        for j in range(len(A[0])): # no. of columns 
             
            row.append(A[i][j] + B[i][j]) # add matrix A and B 
        output.append(row) 
    return output    # return the sum of both matrix 
 
m = int(input("Enter the value of m or Rown")) # take the rows 
n = int(input("Enter the value of n or columnsn")) # take the columns 
 
print("Enter the First matrix ") # print the first matrix 
A = two_d_matrix(m, n) # call the matrix function 
print("display the first(A) matrix") 
print(A) # print the matrix 
 
print("Enter the Second(B) matrix ") 
B = two_d_matrix(m, n) # call the matrix function 
print("display the Second(B) matrix") 
print(B) # print the B matrix 
 
s= sum(A, B) # call the sum function 
print(s) # print the sum of A and B matrix. 

Выход:

1090-23cookie-checkДвумерный массив в Python — основы работы