# Algorithms Exercise 2

## Imports


```python
%matplotlib inline
from matplotlib import pyplot as plt
import seaborn as sns
import numpy as np
```

## Peak finding

Write a function `find_peaks` that finds and returns the indices of the local maxima in a sequence. Your function should:

* Properly handle local maxima at the endpoints of the input array.
* Return a Numpy array of integer indices.
* Handle any Python iterable as input.


```python
def find_peaks(a):
    """Find the indices of the local maxima in a sequence."""
    a = np.array(a)
    s = np.sign(np.diff(a))
    d = np.diff(s)
    ind  = [i for i in range(len(d)) if d[i] == 2]
    if s[-1] == 1:
         ind.append(len(a)-1)
    return(np.array(ind))
    

```


```python
p1 = find_peaks([2,0,1,0,2,0,1])
assert np.allclose(p1, np.array([0,2,4,6]))
p2 = find_peaks(np.array([0,1,2,3]))
assert np.allclose(p2, np.array([3]))
p3 = find_peaks([3,2,1,0])
assert np.allclose(p3, np.array([0]))
```


```python
find_peaks([2,2,2,1,2,2,2])
```




    array([2])



Here is a string with the first 10000 digits of $\pi$ (after the decimal). Write code to perform the following:

* Convert that string to a Numpy array of integers.
* Find the indices of the local maxima in the digits of $\pi$.
* Use `np.diff` to find the distances between consequtive local maxima.
* Visualize that distribution using an appropriately customized histogram.


```python
from sympy import pi, N
pi_digits_str = str(N(pi, 10001))[2:]
```


```python
pi_int = np.array(list(pi_digits_str), dtype="int")
pks = find_peaks(pi_int)
pks_diff = np.diff(pks)
plt.hist(pks_diff, bins = range(0,max(pks_diff)+1))
min(pks_diff), max(pks_diff)
```


```python
pks
```




    array([   1,    4,    7, ..., 9993, 9995, 9999])




```python
assert True # use this for grading the pi digits histogram
```

fb1e278eb1361df546f15d0480a7f058550ef6d7
ipynb
Jupyter Notebook
assignments/assignment07/AlgorithmsEx02.ipynb
aschaffn/PHYS202-2015-work
c0e2eca987ed717a8cc345a7cf556b8111aa5457
MIT
assignments/assignment07/AlgorithmsEx02.ipynb
aschaffn/PHYS202-2015-work
c0e2eca987ed717a8cc345a7cf556b8111aa5457
MIT
assignments/assignment07/AlgorithmsEx02.ipynb
aschaffn/PHYS202-2015-work
c0e2eca987ed717a8cc345a7cf556b8111aa5457
MIT
Qwen/Qwen-72B
1. YES
2. YES