Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- import numpy as np
- def fizzbuzz(number):
- if number % 15 == 0:
- return np.array(["fizzbuzz"], dtype="object")
- elif number % 5 == 0:
- return np.array(["buzz"], dtype="object")
- elif number % 3 == 0:
- return np.array(["fizz"], dtype="object")
- else:
- return np.array([str(number)], dtype="object")
- fizzbuzz_vectorized = np.vectorize(fizzbuzz)
- n = 30
- input_data = np.array([a+1 for a in range(n)])
- fizzbuzz_output = fizzbuzz_vectorized(input_data).ravel()
- fizzbuzz_output
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement