Как построить функцию, определенную с 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() 

Для ответа на то, почему ваш код не работает, когда вы назвали функцию «з», вы не сделали передать любой параметр, следовательно, это возвращает определение функции или указатель местоположения памяти функции. Даже если вы прошли rr до h, h не обрабатывался, чтобы преобразовать его в итерируемый.