write plot legend fix

RUBi-ZA · Jun 8, 2018 · fe78a28 · fe78a28
1 parent 53ab574
commit fe78a28
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/lib/write_plot.py b/lib/write_plot.py
@@ -106,7 +106,7 @@ def write_fig(file_name, pca, out_dir, title):
 	plt.ylabel('PC2')
 	plt.title(title)
 	cbar = plt.colorbar()
-	cbar.set_label('Time [ps]')
+	cbar.set_label('#Frame')
 	fig.savefig(fname)
 
 	## # plot 1 and 3 PC
@@ -117,7 +117,7 @@ def write_fig(file_name, pca, out_dir, title):
 	plt.ylabel('PC3')
 	plt.title(title)
 	cbar = plt.colorbar()
-	cbar.set_label('Time [ps]')
+	cbar.set_label('#Frame')
 	fig.savefig(fname)
 
 	## # plot 2 and 3 PC
@@ -128,7 +128,7 @@ def write_fig(file_name, pca, out_dir, title):
 	plt.ylabel('PC3')
 	plt.title(title)
 	cbar = plt.colorbar()
-	cbar.set_label('Time [ps]')
+	cbar.set_label('#Frame')
 	fig.savefig(fname)
 	return;