如何绘制用 def 定义的函数？

Question

我有一个函数

np.sin(x / 2.) * np.exp(x / 4.) + 6. * np.exp(-x / 4.)

我可以使用以下代码来绘制它：

x = np.arange(-5, 15, 2)
y = np.sin(x / 2.) * np.exp(x / 4.) + 6. * np.exp(-x / 4.)
plt.plot(x, y)
plt.show()

但如果我定义函数绘图不起作用：

rr = np.arange(-5, 15, 2)

def y(o): 
    return np.sin(o / 2.) * np.exp(o / 4.) + 6. * np.exp(-o / 4.)

def h(b):
    return int(y(b))

plt.plot(rr, h)
plt.show()

为什么会发生，如何更改代码来绘制函数？

Answer 1

试试这个：

import numpy as np
import matplotlib.pyplot as plt

rr = np.arange(-5, 15, 2)

def y(o): 
    return np.sin(o / 2.) * np.exp(o / 4.) + 6. * np.exp(-o / 4.)

plt.plot(rr, y(rr).astype(np.int))
plt.show()

Answer 2

匈奴的回答是正确的。

但是，如果您对使用两个函数定义非常具体，那么试试这个：

def y(o): 
    return np.sin(o / 2.) * np.exp(o / 4.) + 6. * np.exp(-o / 4.)
def h(b):
    l = []
    for i in b:
        l.append(int(y(i)))
    return l
rr = np.arange(-5, 15, 2)
plt.plot(rr, h(rr))
plt.show()

要回答为什么您的代码不起作用，当您调用函数“h”时，您没有传递任何参数，因此返回函数定义或函数的内存位置指针。即使您已将 rr 传递给 h，h 也不会被处理以将其转换为可迭代对象。