From 886b39de557b4d5f54f5ca11559fca9799534280 Mon Sep 17 00:00:00 2001
From: Josh Rosen <rosenville@gmail.com>
Date: Fri, 10 Aug 2012 01:10:02 -0700
Subject: Add Python API.

---
 python/tc.py | 22 ++++++++++++++++++++++
 1 file changed, 22 insertions(+)
 create mode 100644 python/tc.py

(limited to 'python')

diff --git a/python/tc.py b/python/tc.py
new file mode 100644
index 0000000000..5dcc4317e0
--- /dev/null
+++ b/python/tc.py
@@ -0,0 +1,22 @@
+from rdd import SparkContext
+
+sc = SparkContext("local", "PythonWordCount")
+e = [(1, 2), (2, 3), (4, 1)]
+
+tc = sc.parallelizePairs(e)
+
+edges = tc.mapPairs(lambda (x, y): (y, x))
+
+oldCount = 0
+nextCount = tc.count()
+
+def project(x):
+    return (x[1][1], x[1][0])
+
+while nextCount != oldCount:
+    oldCount = nextCount
+    tc = tc.union(tc.join(edges).mapPairs(project)).distinct()
+    nextCount = tc.count()
+
+print "TC has %i edges" % tc.count()
+print tc.collect()
-- 
cgit v1.2.3