summaryrefslogtreecommitdiff
path: root/scripts/analysis/error-time.py
blob: 37f0ecec0334d8c336c8140ba04e89b3d6bce57b (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
#!/usr/bin/env python3

import matplotlib.pyplot as plt
import numpy as np
import pandas as pd
import sys

def main(filename, robots):
    s = pd.read_csv(filename).to_numpy().reshape((-1, robots, 2, 2))
    error = s[:,:,1] - s[:,:,0]
    linear = np.sqrt(error[:,:,0] ** 2 + error[:,:,1] ** 2)

    plt.plot(linear)
    plt.ylabel('Error (m)')
    plt.xlabel('Time')
    plt.show()


if __name__ == '__main__':
    if len(sys.argv) != 3:
        sys.exit(1)
    filename = sys.argv[1]
    robots = int(sys.argv[2])
    main(filename, robots)