less invasive error handling

packages/python/pandas.yap

@@ -1,7 +1,61 @@
+%matplotlib inline
+import numpy as np
+import seaborn as sns
+import matplotlib.pyplot as plt
+sns.set(style="white", context="talk")
+rs = np.random.RandomState(7)
 
-:- use_module( library(python) ).
+pos={0:(0,0),
+     1:(1,0),
+     2:(0,1),
+     3:(1,1),
+     4:(0.1,0.9),
+     5:(0.3,1.1),
+     6:(0.9,0.9)
+     }
 
-:- := import(pandas)).
+names={4:'MMM',
+     5:'XXX',
+     6:'ZZZ'}
 
-pred2panda(Pred, Obj) :-
-	
+def plot1(y10,y20):
+    def gen(f,f0):
+        return [f[0],f[1],-f[2]]/max(f,f0)
+    ax1 = plt.subplot2grid((1,2), (0,0), colspan=2)
+    ax2 = plt.subplot2grid((1,2), (0,1), colspan=2)
+    ax3 = plt.subplot2grid((2,2), (2,0), colspan=2, rowspan=2)
+
+    xs = ["+-","++","--"]
+    y1 = gen(y10, y20)
+    sns.barplot(xs, y1, palette="RdBu_r", ax=ax1)
+    y2 = gen(y20,y10)
+    sns.barplot(xs, y2, palette="Set3", ax=ax2)
+    # Finalize the plot
+    # sns.despine(bottom=True)
+
+
+    G=nx.Graph()
+    i=0
+    G.pos={} # location
+    G.pop={} # size
+    lpos={0:(0,0),1:(0,0),2:(0,0),3:(0,0)}
+    last=len(pos)-1
+    for i in range(4,len(pos)):
+        G.pos[i]=pos[i]
+        G.pop[i]=2000
+        (x,y) = pos[i]
+        lpos[i] = (x,y-0.05)
+        if i > 4:
+            G.add_edge(i-1,i)
+        else:
+            G.add_edge(2,i)
+    G.add_edge(3,last)
+    nx.draw_networkx_nodes(G,pos,nodelist=range(4,len(pos)),ax=ax3)
+    nx.draw_networkx_nodes(G,pos,nodelist=[0,1,2,3],node_color='b',ax=ax3)
+    nx.draw_networkx_edges(G,pos,alpha=0.5,ax=ax3)
+    nx.draw_networkx_labels(G,lpos,names,alpha=0.5,ax=ax3)
+    plt.axis('off')
+    plt.tight_layout(h_pad=3)
+    plt.savefig("house_with_colors.png") # save as png
+
+plot1([20,30,10],[30,30,5])