mirror of
https://github.com/Visualize-ML/Book4_Power-of-Matrix.git
synced 2026-02-03 02:24:03 +08:00
22 lines
422 B
Python
22 lines
422 B
Python
|
|
###############
|
|
# Authored by Weisheng Jiang
|
|
# Book 4 | From Basic Arithmetic to Machine Learning
|
|
# Published and copyrighted by Tsinghua University Press
|
|
# Beijing, China, 2022
|
|
###############
|
|
|
|
# Bk4_Ch4_02.py
|
|
|
|
import numpy as np
|
|
|
|
A = np.matrix([[1,2,3],
|
|
[4,5,6],
|
|
[7,8,9]])
|
|
|
|
# extract diagonal elements
|
|
a = np.diag(A)
|
|
|
|
# construct a diagonal matrix
|
|
A_diag = np.diag(a)
|