Made some changes

This commit is contained in:
Cory Balaton 2023-10-22 15:19:45 +02:00
parent 0dfda79fda
commit e42c108d20
No known key found for this signature in database
GPG Key ID: 3E5FCEBFD80F432B

View File

@ -34,13 +34,18 @@ def main():
"seagreen",
"darkred"
]
start_pos = set()
for label, color, file in zip(labels, colors, files):
with open(file) as f:
lines = f.readlines()
t = np.linspace(0, 50, len(lines))
r = np.array([list(map(float, line.strip().split(","))) for line in lines])
plt.plot(r[:,0], r[:,1], label=label, color=color)
start_pos.add((r[0,0],r[0,1]))
for pos in start_pos:
plt.plot(*pos, "o", color="black")
plt.xlabel(r"x $(\mu m)$")
plt.ylabel(r"y $(\mu m)$")
plt.legend(loc="upper right")