'''
Author: SJ2050
Date: 2021-10-27 12:50:53
LastEditTime: 2021-10-27 12:57:38
Version: v0.0.1
Description: Solution for homework 1.2.
Copyright © 2021 SJ2050
'''
import numpy as np

if __name__ == '__main__':
    mat = np.diag(range(1, 5), -1)
    print(mat)