python中rcparams_Python pylab.rcParams方法代碼示例

# 需要導入模塊: import pylab [as 別名]

# 或者: from pylab import rcParams [as 別名]

def plot_rels(data, labels=None, colors=None, outfile="rels", latent=None, alpha=0.8):

ns, n = data.shape

if labels is None:

labels = list(map(str, range(n)))

ncol = 5

# ncol = 4

nrow = int(np.ceil(float(n * (n - 1) / 2) / ncol))

#nrow=1

#pylab.rcParams.update({'figure.autolayout': True})

fig, axs = pylab.subplots(nrow, ncol)

fig.set_size_inches(5 * ncol, 5 * nrow)

#fig.set_canvas(pylab.gcf().canvas)

pairs = list(combinations(range(n), 2)) #[:4]

pairs = sorted(pairs, key=lambda q: q[0]**2+q[1]**2) # Puts stronger relationships first

if colors is not None:

colors = (colors - np.min(colors)) / (np.max(colors) - np.min(colors)).clip(1e-7)

for ax, pair in zip(axs.flat, pairs):

if latent is None:

ax.scatter(data[:, pair[0]], data[:, pair[1]], marker='.', edgecolors='none', alpha=alpha)

else:

# cs = 'rgbcmykrgbcmyk'

markers = 'x+.o,<>^^<>,+x.'

for j, ind in enumerate(np.unique(latent)):

inds = (latent == ind)

ax.scatter(data[inds, pair[0]], data[inds, pair[1]], c=colors[inds], cmap=pylab.get_cmap("jet"),

marker=markers[j], alpha=0.5, edgecolors='none', vmin=0, vmax=1)

ax.set_xlabel(shorten(labels[pair[0]]))

ax.set_ylabel(shorten(labels[pair[1]]))

for ax in axs.flat[axs.size - 1:len(pairs) - 1:-1]:

ax.scatter(data[:, 0], data[:, 1], marker='.')

pylab.rcParams['font.size'] = 12 #6

pylab.draw()

#fig.set_tight_layout(True)

fig.tight_layout()

for ax in axs.flat[axs.size - 1:len(pairs) - 1:-1]:

ax.set_visible(False)

filename = outfile + '.png'

if not os.path.exists(os.path.dirname(filename)):

os.makedirs(os.path.dirname(filename))

fig.savefig(outfile + '.png') #df')

pylab.close('all')

return True