E6.1A "comb" Function
To create a "comb'' of values in an array of length \(N\) for which every \(n\)th element is one but with zeros everywhere else:
2_F_Ea-combfunction
import numpy as np
N, n = 101, 5
def f(i):
return (i % n == 0) * 1
comb = np.fromfunction(f, (N,), dtype=int)
print(comb)
The output is:
[1 0 0 0 0 1 0 0 0 0 1 0 0 0 0 1 0 0 0 0 1 0 0 0 0 1 0 0 0 0 1 0 0 0 0 1 0
0 0 0 1 0 0 0 0 1 0 0 0 0 1 0 0 0 0 1 0 0 0 0 1 0 0 0 0 1 0 0 0 0 1 0 0 0
0 1 0 0 0 0 1 0 0 0 0 1 0 0 0 0 1 0 0 0 0 1 0 0 0 0 1]