def plot(self, **kwargs):
if len(self.all_vars) == 1 and len(self.free_rvs) == 1:
pfun = FunDistr(self.nddistr, breakPoints = self.nddistr.Vars[0].range())
pfun.plot(label = str(self.nddistr.Vars[0].getSymname()), **kwargs)
legend()
elif len(self.all_vars) == 2 and len(self.free_rvs) == 2:
plot_2d_distr(self.nddistr, **kwargs)
elif len(self.all_vars) == 2 and len(self.free_rvs) == 1:
a, b = self.free_rvs[0].range()
freesym = self.free_rvs[0].getSym()
fun = my_lambdify(freesym, self.rv_to_equation[self.dep_rvs[0]], "numpy")
ax = plot_1d1d_distr(self.nddistr, a, b, fun)
ax.set_xlabel(self.free_rvs[0].getSymname())
ax.set_ylabel(self.dep_rvs[0].getSymname())
else:
raise RuntimeError("Too many variables.")
def plot(self, **kwargs):
if len(self.all_vars) == 1 and len(self.free_rvs) == 1:
pfun = FunDistr(self.nddistr,
breakPoints=self.nddistr.Vars[0].range())
pfun.plot(label=str(self.nddistr.Vars[0].getSymname()), **kwargs)
legend()
elif len(self.all_vars) == 2 and len(self.free_rvs) == 2:
plot_2d_distr(self.nddistr, **kwargs)
elif len(self.all_vars) == 2 and len(self.free_rvs) == 1:
a, b = self.free_rvs[0].range()
freesym = self.free_rvs[0].getSymname()
fun = my_lambdify([freesym], self.rv_to_equation[self.dep_rvs[0]],
"numpy")
ax = plot_1d1d_distr(self.nddistr, a, b, fun)
ax.set_xlabel(self.free_rvs[0].getSymname())
ax.set_ylabel(self.dep_rvs[0].getSymname())
elif len(self.all_vars) == 1 and len(self.free_rvs) == 0:
DiracSegment(self.as_const(), 1).plot(**kwargs)
else:
raise RuntimeError("Too many variables.")
