62 lines
1.5 KiB
Prolog
62 lines
1.5 KiB
Prolog
%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)
|
|
|
|
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)
|
|
}
|
|
|
|
names={4:'MMM',
|
|
5:'XXX',
|
|
6:'ZZZ'}
|
|
|
|
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])
|