Numpy Matrix Addition Code

Previous
Next

Read elements into 2 dimensional array:

import numpy as np

m = int(input("Enter row size :"))
n = int(input("Enter column size :"))
matrix = np.ndarray(shape=(m,n),dtype=int)

print("Size :",matrix.size)
print("shape :",matrix.shape)
print("Dimension :",matrix.ndim)
print("Datatype :",matrix.dtype)

Read elements into matrix:

import numpy as np
print("Enter Array dimension :")
m = int(input("Rows :"))
n = int(input("Columns :"))
matrix = np.ndarray(shape=(m,n), dtype=int)

print("Enter %d elements of %dx%d matrix :" %(m*n,m,n))
for i in range(m):
    for j in range(n):
        matrix[i][j] = int(input())

print("Matrix is :")
print(matrix)

Matrix addition:

import numpy as np
A = np.ndarray(shape=(3,3),dtype=int)
val=1
for i in range(3):
    for j in range(3):
        A[i][j] = val
        val=val+1
print(A)

B = np.ndarray(shape=(3,3),dtype=int)
val=9
for i in range(3):
    for j in range(3):
        B[i][j] = val
        val=val-1
print(B)

C = np.ndarray(shape=(3,3),dtype=int)
for i in range(3):
    for j in range(3):
        C[i][j] = A[i][j]+B[i][j]

print("Added matrix :")
print(C)
Previous
Next

Add Comment

Courses Enquiry Form