在python中绘制数学函数

Question

我正在尝试编写称为绘图的函数，该函数接受i / p参数Z，p和q并绘制函数

f(y) = det(Z − yI) on the interval [p, q]

（注意：我是单位矩阵。）det（）是行列式。

为了找到det（），可以使用numpy.linalg.det() ；对于身份矩阵，可以使用np.matlib.identity(n)

有没有办法在python中编写这样的函数？ 并绘制它们？

import numpy as np

def f(y):
    I2 = np.matlib.identity(y)
    x = Z-yI2
    numpy.linalg.det(x)
    ....

我的试穿正确吗？ 还有其他选择吗？

Answer 1

您可以使用以下实现。

import numpy as np
import matplotlib.pyplot as plt

def f(y, Z):  
    n, m = Z.shape
    assert(n==m)
    I = np.identity(n)
    x = Z-y*I
    return np.linalg.det(x)

Z = np.matrix('1 2; 3 4')
p = -15
q = 15
y = np.linspace(p, q)
w = np.zeros(y.shape)

for i in range(len(y)):
    w[i] = f(y[i], Z)

plt.plot(y, w)
plt.show()