From 99bb43a988cf0c5d30cf61d0c11f99b09ff6b5d4 Mon Sep 17 00:00:00 2001
From: Samo Penic <samo.penic@gmail.com>
Date: Thu, 05 Jan 2017 20:57:34 +0000
Subject: [PATCH] Added getColumn() and getTable() functions to Statistics class

---
 trisurf/trisurf.py |   29 +++++++++++++++++++++++++++++
 1 files changed, 29 insertions(+), 0 deletions(-)

diff --git a/trisurf/trisurf.py b/trisurf/trisurf.py
index e386595..6eae40a 100644
--- a/trisurf/trisurf.py
+++ b/trisurf/trisurf.py
@@ -258,6 +258,35 @@
 		fields=shlex.split(lines[1])
 		epoch2=fields[0]
 		return (fields[1])
+
+	def getColumn(self,n=0):
+		lines=self.readText().splitlines()
+		col=[]
+		name=lines[0].split()[n]
+		for line in lines:
+			col.append(line.split()[n])
+		return col[2:],name
+
+	def getTable(self):
+		lines=self.readText().splitlines()
+		keys=lines[0].split()
+		table={}
+		def num(s):
+			try:
+				return int(s)
+			except ValueError:
+				return float(s)
+		for key in keys:
+			table[key]=[]
+		for line in lines:
+			fields=line.split()
+			for i in range(0,len(keys)-1):
+				try:
+					value=num(fields[i])
+					table[keys[i]].append(value)
+				except:
+					pass
+		return table	
 		
 
 	def readText(self):

--
Gitblit v1.9.3