Add show values inner function

minjk-bl · minjk-bl · commit b44bfff03e2a · 2022-06-16T12:08:31.000+09:00
diff --git a/python/userCommand.py b/python/userCommand.py
@@ -89,4 +89,42 @@ def vp_plot_feature_importances(model, X_train=None, sort=False):
     _vp_plt.xlabel('Feature importance Percentage')
     _vp_plt.ylabel('Features')
                         
-    _vp_plt.show()
+    _vp_plt.show()
+######
+# Visual Python: Visualization > Seaborn
+######
+def vp_seaborn_show_values(axs, precision=1, space=0.01):
+    pstr = '{:.' + str(precision) + 'f}'
+    
+    def _single(ax):
+        # check orient
+        orient = 'v'
+        if len(ax.patches) == 1:
+            # check if 0
+            if ax.patches[0].get_x() == 0:
+                orient = 'h'
+        else:
+            # compare 0, 1 patches
+            p0 = ax.patches[0]
+            p1 = ax.patches[1]
+            if p0.get_x() == p1.get_x():
+                orient = 'h'
+                
+        if orient == 'v':
+            for p in ax.patches:
+                _x = p.get_x() + p.get_width() / 2
+                _y = p.get_y() + p.get_height() + (p.get_height()*space)
+                value = pstr.format(p.get_height())
+                ax.text(_x, _y, value, ha='center') 
+        elif orient == 'h':
+            for p in ax.patches:
+                _x = p.get_x() + p.get_width() + (space - 0.01)
+                _y = p.get_y() + p.get_height() / 2
+                value = pstr.format(p.get_width())
+                ax.text(_x, _y, value, ha='left')
+
+    if isinstance(axs, _vp_np.ndarray):
+        for idx, ax in _vp_np.ndenumerate(axs):
+            _single(ax)
+    else:
+        _single(axs)
