KPMG interview question

oops concept program to find the diogonal of a metrix python basic questions