assets

CH17/data/cities_ranking.csv

@@ -0,0 +1,3 @@
+rank,city,score,A,B,C,D,E,F
+1,BJ,93.74,86.98,100.0,95.88,100.0,98.05,82.62
+2,SH,88.23,88.96,90.81,76.25,90.25,100.0,75.72

CH17/utils.py

@@ -0,0 +1,43 @@
+#! /usr/bin/env python
+
+# Project:  Lihang
+# Filename: utils
+# Date: 6/12/19
+# Author: 😏 <smirk dot cao at gmail dot com>
+import numpy as np
+import matplotlib.pyplot as plt
+
+
+class Radar(object):
+    def __init__(self, feas=None, labels=None):
+        self.feas = feas
+        self.colors = ['green', 'blue', 'red', 'yellow', 'black']
+        self.labels = labels
+
+    def plot(self, data):
+        assert self.feas is not None
+
+        angles = np.linspace(0.1 * np.pi, 2.1 * np.pi, len(self.feas),
+                             endpoint=False)
+        angles = np.concatenate((angles, [angles[0]]))
+
+        fig = plt.figure(figsize=(5, 5))
+        ax = fig.add_subplot(111, polar=True)
+
+        # ax.legend(loc=[0.25, 1.15], fontsize=18)
+        ax.set_yticklabels([])
+        ax.set_thetagrids(angles * 180/np.pi, self.feas, fontsize=12)
+        ax.grid(True)
+
+        for idx, label in enumerate(self.labels):
+            stats = data[idx]
+            stats = np.concatenate((stats, [stats[0]]))
+            ax.plot(angles, stats, '--', linewidth=1, 
+                    c=self.colors[idx % len(self.colors)],
+                    label=str(label))
+            ax.fill(angles, stats, 
+                    c=self.colors[idx % len(self.colors)],
+                    alpha=0.25)
+        plt.legend()
+        plt.show()
+        return fig