import csv def load_2d_experiment(filename: str): with open(filename) as csvfile: spamreader = csv.reader(csvfile, delimiter=',', quotechar='|') rows = list(spamreader) x = [int(row[0]) for row in rows] y = [float(row[1]) for row in rows] return (x, y) def load_3d_experiment(filename: str): with open(filename) as csvfile: offset = 1 if csv.Sniffer().has_header(csvfile.read(1024)) else 0 csvfile.seek(0) spamreader = csv.reader(csvfile, delimiter=',', quotechar='|') rows = list(spamreader)[offset:] x = [] y = [] z = [] # for row in rows[offset]: # print(row) # x.append(int(row[0])) # y.append(int(row[1])) # z.append(float(row[2])) x = [int(row[0]) for row in rows] y = [int(row[1]) for row in rows] z = [float(row[2]) for row in rows] return (x, y, z)