Mesh Bio interview question

Python program to print out identity matrix