Add legend to rsgislib.tools.plotting.plot_vec_fields plots.

remotesensinginfo · Jan 24, 2024 · 0fb2f19 · 0fb2f19
1 parent 8983cdd
commit 0fb2f19
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/python/rsgislib/tools/plotting.py b/python/rsgislib/tools/plotting.py
@@ -727,13 +727,15 @@ def plot_vec_fields(
             feat_y_vals = y_vals
 
         if plt_line:
-            ax.plot(feat_x_vals, feat_y_vals)
+            ax.plot(feat_x_vals, feat_y_vals, label=f"Feature {feat_id}")
         else:
-            ax.scatter(feat_x_vals, feat_y_vals)
+            ax.scatter(feat_x_vals, feat_y_vals, label=f"Feature {feat_id}")
 
     ax.set_xlabel(x_lbl)
     ax.set_ylabel(y_lbl)
     ax.set_title(title)
+    if feat_id_field is not None:
+        plt.legend()
     fig.tight_layout()
     plt.savefig(out_plot_file)