28 lines
825 B
Python
28 lines
825 B
Python
import csv
|
|
import sys
|
|
|
|
import matplotlib.pyplot as plt
|
|
|
|
data = []
|
|
|
|
file = 1
|
|
try:
|
|
while True:
|
|
print("{}/heatmap_congestion_by_coordinate_{}.csv".format(sys.argv[1], file))
|
|
with open("{}/heatmap_congestion_by_coordinate_{}.csv".format(sys.argv[1], file)) as f:
|
|
file_data = []
|
|
reader = csv.reader(f, delimiter=',')
|
|
for row in reader:
|
|
file_data.append([float(x) for x in row if x != ''])
|
|
data.append(file_data)
|
|
file += 1
|
|
except FileNotFoundError:
|
|
pass
|
|
|
|
for i, file_data in enumerate(data):
|
|
plt.imshow(file_data, cmap="gray", aspect="equal")
|
|
plt.title("heatmap for iteration {}".format(i+1))
|
|
plt.tight_layout()
|
|
plt.savefig("{}/heatmap_congestion_by_coordinate_{:03}.png".format(sys.argv[1], i), dpi=300)
|
|
plt.clf()
|