I need the code to make this modification possible.

Also, the code must be done in python

4. Zeroing out terms outside a band The matrix A shown below is fully populated with non-zero terms. Your task is to write a function that takes a square matrix and zeros out the terms on each row that lie beyond a specified distance from the diagonal. The maximum number of non-zero terms on any row will be 2HBW+1. You will use your code with the following matrix and use HBW=3. To make sure you understand what the matrix should look like, I have also included what it should look like after your function modifies it