def readCSV(filepath): reader = CSVReader(FileReader(filepath), ",") ls = reader.readAll() data = ArrayList() for item in ls: data.add(item) return data
def readKeyCSV(filepath):
reader = CSVReader(FileReader(filepath), ",")
ls = reader.readAll()
keymap = HashMap()
for item in ls:
keymap.put(int(item[0]), int(item[1]))
print item[0], item[1]
return keymap
def readCSV(filepath):
reader = CSVReader(FileReader(filepath), "\t")
ls = reader.readAll()
f, x, y, z = [], [], [], []
for i in range(len(ls)):
if i is not 0:
f.append(ls.get(i)[1])
x.append(ls.get(i)[4])
y.append(ls.get(i)[5])
z.append(ls.get(i)[6])
# print x[i-1], y[i-1], z[i-1]
return f, x, y, z
def readDataCSV(fp):
reader = CSVReader(FileReader(fp), ",")
ls = reader.readAll()
tracks = HashMap()
for idx, item in enumerate(ls):
if idx > 0:
if tracks.containsKey(int(item[0])):
pass
else:
track = [[], [], [], []]
tracks.put(int(item[0]), track) # trackID in str.
track = tracks.get(int(item[0]))
track[0].append(int(item[1])) # frame 0
track[1].append(item[3]) # x 1
track[2].append(item[4]) # y 2
roi = Roi(int(item[5]), int(item[6]), int(item[7]), int(item[8]))
track[3].append(roi) # a roi 3
return tracks
