28 lines
No EOL
951 B
Python
28 lines
No EOL
951 B
Python
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) |