Advertisement
Guest User

Untitled

a guest
Aug 22nd, 2017
69
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 0.51 KB | None | 0 0
  1. import numpy as np
  2.  
  3. def fizzbuzz(number):
  4. if number % 15 == 0:
  5. return np.array(["fizzbuzz"], dtype="object")
  6. elif number % 5 == 0:
  7. return np.array(["buzz"], dtype="object")
  8. elif number % 3 == 0:
  9. return np.array(["fizz"], dtype="object")
  10. else:
  11. return np.array([str(number)], dtype="object")
  12.  
  13. fizzbuzz_vectorized = np.vectorize(fizzbuzz)
  14.  
  15. n = 30
  16.  
  17. input_data = np.array([a+1 for a in range(n)])
  18.  
  19. fizzbuzz_output = fizzbuzz_vectorized(input_data).ravel()
  20.  
  21. fizzbuzz_output
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement