diff options
Diffstat (limited to 'examples/src/main/python/pi.py')
-rwxr-xr-x | examples/src/main/python/pi.py | 4 |
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() |