import numpy as np
%matplotlib inline
import matplotlib.pyplot as plt
import seaborn as sns

"""
Explanation: Numpy Exercise 4
Imports
End of explanation
"""


import networkx as nx
K_5=nx.complete_graph(5)
nx.draw(K_5)

"""
Explanation: Complete graph Laplacian
In discrete mathematics a Graph is a set of vertices or nodes that are connected to each other by edges or lines. If those edges don't have directionality, the graph is said to be undirected. Graphs are used to model social and communications networks (Twitter, Facebook, Internet) as well as natural systems such as molecules.
A Complete Graph, $K_n$ on $n$ nodes has an edge that connects each node to every other node.
Here is $K_5$:
End of explanation
"""


def complete_deg(n):
    """Return the integer valued degree matrix D for the complete graph K_n."""
    tits = np.zeros((n,n), dtype=int)
    for x in range(n):
        tits[x,x] = n-1
    return tits
    

D = complete_deg(5)
assert D.shape==(5,5)
assert D.dtype==np.dtype(int)
assert np.all(D.diagonal()==4*np.ones(5))
assert np.all(D-np.diag(D.diagonal())==np.zeros((5,5),dtype=int))

"""
Explanation: The Laplacian Matrix is a matrix that is extremely important in graph theory and numerical analysis. It is defined as $L=D-A$. Where $D$ is the degree matrix and $A$ is the adjecency matrix. For the purpose of this problem you don't need to understand the details of these matrices, although their definitions are relatively simple.
The degree matrix for $K_n$ is an $n \times n$ diagonal matrix with the value $n-1$ along the diagonal and zeros everywhere else. Write a function to compute the degree matrix for $K_n$ using NumPy.
End of explanation
"""


def complete_adj(n):
    """Return the integer valued adjacency matrix A for the complete graph K_n."""
    hellyeah = np.ones((n,n), dtype=int)
    for ilovepython in range(n):
        hellyeah[ilovepython,ilovepython] = 0
    return hellyeah

A = complete_adj(5)
assert A.shape==(5,5)
assert A.dtype==np.dtype(int)
assert np.all(A+np.eye(5,dtype=int)==np.ones((5,5),dtype=int))

"""
Explanation: The adjacency matrix for $K_n$ is an $n \times n$ matrix with zeros along the diagonal and ones everywhere else. Write a function to compute the adjacency matrix for $K_n$ using NumPy.
End of explanation
"""


def L(n):
    return complete_deg(n)-complete_adj(n)

Yay = np.empty((98,))

n = np.linspace(2, 100, 98)

for z in range(2, 98):
        lap = L(z)
        lapeig = np.linalg.eigvals(lap)
        Yay[z] = np.min(lapeig)

plt.plot(n, Yay)

"""
Explanation: Use NumPy to explore the eigenvalues or spectrum of the Laplacian L of $K_n$. What patterns do you notice as $n$ changes? Create a conjecture about the general Laplace spectrum of $K_n$.
End of explanation
"""


