YOPO

Controller/src/so3_control/logger/log_plot.py

@@ -0,0 +1,28 @@
+import sys
+import numpy as np
+import matplotlib.pyplot as plt
+
+if __name__ == '__main__':
+    file_path = sys.argv[1]
+    temp = np.loadtxt(file_path, dtype=str, delimiter=",")
+    print('load file:', file_path)
+    label = temp[0, :].astype(str)
+    data = temp[1:, :].astype(np.float64)
+    timestamp = (data[:, 0] - data[0, 0])
+    print('data length:', len(data))
+    # print(timestamp)
+    while 1:
+        for i in range(len(label)):
+            if i != 0:
+                print('[', i, ']: ', label[i])
+        print('plot the curve')
+        print('example: 1 2 3')
+        wtp = input('want to plot(\'q\' to quit):')
+        if wtp == 'q':
+            break
+        wtp = [int(n) for n in wtp.split()]
+        for i in range(len(wtp)):
+            plt.plot(timestamp, data[:, wtp[i]], linewidth=2.5, linestyle='-', marker='.', label=label[wtp[i]])
+        plt.xlabel('t')
+        plt.legend()
+        plt.show()