Added Matplotlib and Comprehensions files
This commit is contained in:
90
Matplotlib/pyplot.py
Normal file
90
Matplotlib/pyplot.py
Normal file
@@ -0,0 +1,90 @@
|
||||
import numpy as np
|
||||
import pandas as pd
|
||||
import matplotlib.pyplot as plt
|
||||
|
||||
# 1. simple plot with 4 numbers
|
||||
plt.plot([1, 3, 2, 4])
|
||||
plt.show()
|
||||
|
||||
# 2. points have x and y values; add title and axis labels
|
||||
plt.plot([1, 2, 3, 4], [1, 4, 9, 16])
|
||||
plt.title('Test Plot', fontsize=8, color='g')
|
||||
plt.xlabel('number n')
|
||||
plt.ylabel('n^2')
|
||||
plt.show()
|
||||
|
||||
# 3. change figure size. plot red dots; set axis scales x: 0-6 and y: 0-20
|
||||
plt.figure(figsize=(1,5)) # 1 inch wide x 5 inches tall
|
||||
plt.plot([1, 2, 3, 4], [1, 4, 9, 16], 'ro') # red-o
|
||||
plt.axis([0, 6, 0, 20]) # [xmin, xmax, ymin, ymax]
|
||||
plt.annotate('square it', (3,6))
|
||||
plt.show()
|
||||
|
||||
# 4. bar chart with four bars
|
||||
plt.clf() # clear figure
|
||||
x = np.arange(4)
|
||||
y = [8.8, 5.2, 3.6, 5.9]
|
||||
plt.xticks(x, ('Ankit', 'Hans', 'Joe', 'Flaco'))
|
||||
plt.bar(x, y)
|
||||
# plt.bar(x, y, color='y')
|
||||
# plt.bar(x, y, color=['lime', 'r', 'k', 'tan'])
|
||||
plt.show()
|
||||
|
||||
# 5. two sets of 10 random dots plotted
|
||||
d = {'Red O' : np.random.rand(10),
|
||||
'Grn X' : np.random.rand(10)}
|
||||
df = pd.DataFrame(d)
|
||||
df.plot(style=['ro','gx'])
|
||||
plt.show()
|
||||
|
||||
# 6. time series - six months of random floats
|
||||
ts = pd.Series(np.random.randn(180), index=pd.date_range('1/1/2018', periods=180))
|
||||
df = pd.DataFrame(np.random.randn(180, 3), index=ts.index, columns=list('ABC'))
|
||||
df.cumsum().plot()
|
||||
plt.show()
|
||||
|
||||
# 7. random dots in a scatter
|
||||
N = 50
|
||||
x = np.random.rand(N)
|
||||
y = np.random.rand(N)
|
||||
colors = np.random.rand(N)
|
||||
sizes = (30 * np.random.rand(N))**2 # 0 to 15 point radii
|
||||
plt.scatter(x, y, s=sizes, c=colors, alpha=0.5)
|
||||
plt.show()
|
||||
|
||||
# 8. load csv file and show multiple chart types
|
||||
df = pd.read_csv('Fremont_weather.txt')
|
||||
print(df)
|
||||
plt.bar(df['month'], df['record_high'], color='r')
|
||||
plt.bar(df['month'], df['record_low'], color='c')
|
||||
plt.plot(df['month'], df['avg_high'], color='k')
|
||||
plt.plot(df['month'], df['avg_low'], color='b')
|
||||
plt.legend() # or plt.figlegend for legend outside the plot area
|
||||
plt.show()
|
||||
|
||||
# 9. subplots
|
||||
fig = plt.figure()
|
||||
fig.suptitle('My SubPlots')
|
||||
fig.add_subplot(221) #top left
|
||||
plt.plot([np.log(n) for n in range(1,10)])
|
||||
fig.add_subplot(222, facecolor='y') #top right
|
||||
fig.add_subplot(223) #bottom left
|
||||
fig.add_subplot(224) #bottom right
|
||||
plt.show()
|
||||
|
||||
fig, plots = plt.subplots(2, sharex=True)
|
||||
fig.suptitle('Sharing X axis')
|
||||
x = range(0,200,5)
|
||||
y = [n**0.8 for n in x]
|
||||
plots[0].plot(x, y, color='r')
|
||||
plots[1].scatter(x, y)
|
||||
|
||||
# 10. save figure to image file
|
||||
plt.figure(figsize=(4,3), dpi=100)
|
||||
plt.plot([245, 170, 148, 239, 161, 196, 112, 258])
|
||||
plt.axis([0, 7, 0, 300])
|
||||
plt.title('Flight Data')
|
||||
plt.xlabel('Speed')
|
||||
# plt.savefig('Flights.png')
|
||||
plt.show()
|
||||
|
||||
Reference in New Issue
Block a user