aboutsummaryrefslogtreecommitdiff
path: root/examples/src/main/python/pi.py
diff options
context:
space:
mode:
Diffstat (limited to 'examples/src/main/python/pi.py')
-rwxr-xr-xexamples/src/main/python/pi.py4
1 files changed, 1 insertions, 3 deletions
diff --git a/examples/src/main/python/pi.py b/examples/src/main/python/pi.py
index b39d710540..e3f0c4aeef 100755
--- a/examples/src/main/python/pi.py
+++ b/examples/src/main/python/pi.py
@@ -32,8 +32,6 @@ if __name__ == "__main__":
.appName("PythonPi")\
.getOrCreate()
- sc = spark.sparkContext
-
partitions = int(sys.argv[1]) if len(sys.argv) > 1 else 2
n = 100000 * partitions
@@ -42,7 +40,7 @@ if __name__ == "__main__":
y = random() * 2 - 1
return 1 if x ** 2 + y ** 2 < 1 else 0
- count = sc.parallelize(range(1, n + 1), partitions).map(f).reduce(add)
+ count = spark.sparkContext.parallelize(range(1, n + 1), partitions).map(f).reduce(add)
print("Pi is roughly %f" % (4.0 * count / n))
spark.stop()