Блог

В настоящее время у меня есть три синусоидальные функции (y1, y2, y3), и я хотел бы суммировать выходные данные функций в новой функции (ytotal), но только там, где выходные данные синусоидальных функций больше 0.

 import numpy as np
import matplotlib.pyplot as plt
#%%

phi = np.linspace(-2*np.pi, 2*np.pi, 100)

y1 = 0.2*np.sin(phi)
y2 = 0.2*np.sin(phi-(120*(np.pi/180)))
y3 = 0.2*np.sin(phi-(240*(np.pi/180)))

#if y1 or y2 or y3 > 0:
#    ytotal = y1 y2 y3


    


plt.plot(phi,y1, label = "Piston 1")
plt.plot(phi,y2, label = "Piston 2")
plt.plot(phi,y3, label = "Piston 3")
#plt.plot(phi,ytotal, label = "Total output")
positions = (0,np.pi/3,2*np.pi/3,np.pi,4*np.pi/3,5*np.pi/3,2*np.pi)
labels = ("0","60","120","180","240","300","360")
plt.xticks(positions, labels)
plt.xlabel('Angular displacement')
plt.ylabel('Stroke')
plt.legend()
plt.show()
  

Результат должен быть примерно следующим:

Вы имеете в виду:

 plt.plot(phi, y1.clip(0) y2.clip(0) y3.clip(0), label='Total')
  

Вывод: