import matplotlib.pyplot as plt
import math

import numpy as np

X = range(100)
Y = [value ** 2 for value in X]

plt.plot(X, Y)
plt.show()

# 曲线
T = range(100)

X = [(2 * math.pi * t) / len(T) for t in T]
Y = [math.sin(value) for value in X]

plt.plot(X, Y)
plt.show()

X = np.linspace(0, 2 * np.pi, 100)
Y = np.sin(X)

plt.plot(X, Y)
plt.show()

X = np.linspace(-3, 2, 200)
Y = X ** 2 - 2 * X + 1.
plt.plot(X, Y)
plt.show()

X = np.linspace(0, 2 * np.pi, 100)
Ya = np.sin(X)
Yb = np.cos(X)

plt.plot(X, Ya)
plt.plot(X, Yb)

plt.show()

data = np.random.rand(1024, 2)
plt.scatter(data[:,0], data[:,1])
plt.show()

data = [5., 25., 50., 20.]
plt.bar(range(len(data)), data)
plt.show()

data = [5., 25., 50., 20.]
plt.bar(range(len(data)), data, width = 1.)
plt.show()

data = [5., 25., 50., 20.]
plt.barh(range(len(data)), data)
plt.show()

data = [[5., 25., 50., 20.],
     [4., 23., 51., 17.],
     [6., 22., 52., 19.]]

X = np.arange(4)

plt.bar(X + 0.00, data[0], color = 'b', width = 0.25)
plt.bar(X + 0.25, data[1], color = 'g', width = 0.25)
plt.bar(X + 0.50, data[2], color = 'r', width = 0.25)

plt.show()

data = [[5., 25., 50., 20.],
     [4., 23., 51., 17.],
     [6., 22., 52., 19.]]

color_list = ['b', 'g', 'r']

gap = .8 / len(data)

for i, row in enumerate(data):
 X = np.arange(len(row))
 plt.bar(X + i * gap, row,
   width = gap,
   color = color_list[i % len(color_list)])
plt.show()

A = [5., 30., 45., 22.]
B = [5., 25., 50., 20.]
X = range(4)
plt.bar(X, A, color = 'b')
plt.bar(X, B, color = 'r', bottom = A)
plt.show()

A = np.array([5., 30., 45., 22.])
B = np.array([5., 25., 50., 20.])
C = np.array([1.,  2.,  1.,  1.])

X = np.arange(4)

plt.bar(X, A, color = 'b')
plt.bar(X, B, color = 'g', bottom = A)
plt.bar(X, C, color = 'r', bottom = A + B)

plt.show()

data = np.array([[5., 30., 45., 22.],
 [5., 25., 50., 20.],
 [1.,  2.,  1.,  1.]])

color_list = ['b', 'g', 'r']

color_list = ['b', 'g', 'r']

X = np.arange(data.shape[1])
for i in range(data.shape[0]):
 plt.bar(X, data[i],
   bottom = np.sum(data[:i], axis = 0),
   color = color_list[i % len(color_list)])
                
plt.show()

women_pop = np.array([5., 30., 45., 22.])
men_pop     = np.array( [5., 25., 50., 20.])

X = np.arange(4)
plt.barh(X, women_pop, color = 'r')
plt.barh(X, -men_pop, color = 'b')
plt.show()

data = [5, 25, 50, 20]
plt.pie(data)
plt.show()

#直方图

X = np.random.randn(1000)
plt.hist(X, bins = 20)
plt.show()

data = np.random.randn(100)
plt.boxplot(data)
plt.show()

data = np.random.randn(100, 5)
plt.boxplot(data)
plt.show()

import matplotlib.tri as tri

data = np.random.rand(100, 2)
triangles = tri.Triangulation(data[:,0], data[:,1])
plt.triplot(triangles)
plt.show()