pi.py 1.2 KB

123456789101112131415161718192021222324252627282930313233343536
  1. #!/usr/bin/env python
  2. # Licensed to Cloudera, Inc. under one
  3. # or more contributor license agreements. See the NOTICE file
  4. # distributed with this work for additional information
  5. # regarding copyright ownership. Cloudera, Inc. licenses this file
  6. # to you under the Apache License, Version 2.0 (the
  7. # "License"); you may not use this file except in compliance
  8. # with the License. You may obtain a copy of the License at
  9. #
  10. # http://www.apache.org/licenses/LICENSE-2.0
  11. #
  12. # Unless required by applicable law or agreed to in writing, software
  13. # distributed under the License is distributed on an "AS IS" BASIS,
  14. # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  15. # See the License for the specific language governing permissions and
  16. # limitations under the License.
  17. from __future__ import print_function
  18. from builtins import range
  19. import pyspark
  20. import random
  21. if not 'sc' in globals():
  22. sc = pyspark.SparkContext()
  23. NUM_SAMPLES = 1000
  24. def sample(p):
  25. x,y = random.random(),random.random()
  26. return 1 if x*x + y*y < 1 else 0
  27. count = sc.parallelize(range(0, NUM_SAMPLES)) \
  28. .map(sample) \
  29. .reduce(lambda a, b: a + b)
  30. print("Pi is roughly %f" % (4.0 * count / NUM_SAMPLES))