pyFTS/partitioners/Util.py

22 lines
658 B
Python
Raw Normal View History

import numpy as np
import pandas as pd
import matplotlib as plt
import matplotlib.colors as pltcolors
import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d import Axes3D
from pyFTS.common import Membership
def plotSets(data, sets,num):
fig = plt.figure(figsize=[20, 10])
maxx = max(data)
minx = min(data)
h = 1/num
for k in range(num):
ax0 = fig.add_axes([0, (k+1)*h, 0.65, h]) # left, bottom, width, height
ax0.set_ylim([0, 1])
ax0.set_xlim([minx, maxx])
for s in sets[k]:
if s.mf == Membership.trimf:
ax0.plot([s.parameters[0],s.parameters[1],s.parameters[2]],[0,1,0])