Question

How to print the entire row and column of a specific cell in a matrix with Python

Answer 1

import numpy as np
 
matrix = np.array([ [1, 2, 3, 4, 5],
                    [0, 3, 6, 7, 8],
                    [9, 9, 6, 5, 4],
                    [8, 8, 8, 1, 1],
                    [2, 1, 1, 3, 5] ])
           
cell_row = 2
cell_col = 3
 
print(matrix[cell_row,: ])
print(matrix[:, cell_col])
 
 
 
'''
run:
 
[9 9 6 5 4]
[4 7 5 1 3]
 
'''

 

Learn SQL from A to Z at LearnSQL.com

Answer 2

def getColumn(matrix, i):
    return [row[i] for row in matrix]

matrix = [ [1, 2, 3, 4, 5],
           [0, 3, 6, 7, 8],
           [9, 9, 6, 5, 4],
           [8, 8, 8, 1, 1],
           [2, 1, 1, 3, 5] ]
          
cell_row = 2
cell_col = 3

print(matrix[cell_row])
print(getColumn(matrix, cell_col))



'''
run:

[9 9 6 5 4]
[4 7 5 1 3]

'''

 

Learn SQL from A to Z at LearnSQL.com