Thanks to visit codestin.com
Credit goes to github.com

Skip to content

Commit fd17060

Browse files
QuLogicQuLogic
committed
Cleanup examples for PEP8 and some style.
Try and use `plt.subplots` more and remove extraneous `plt.draw`.
1 parent 73d041d commit fd17060

62 files changed

Lines changed: 232 additions & 334 deletions

File tree

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

‎examples/axes_grid1/demo_colorbar_of_inset_axes.py‎

Lines changed: 7 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -9,17 +9,17 @@
99
from mpl_toolkits.axes_grid1.inset_locator import inset_axes, zoomed_inset_axes
1010
from mpl_toolkits.axes_grid1.colorbar import colorbar
1111

12+
1213
def get_demo_image():
1314
from matplotlib.cbook import get_sample_data
1415
import numpy as np
1516
f = get_sample_data("axes_grid/bivariate_normal.npy", asfileobj=False)
1617
z = np.load(f)
1718
# z is a numpy array of 15x15
18-
return z, (-3,4,-4,3)
19+
return z, (-3, 4, -4, 3)
1920

2021

21-
fig = plt.figure(1, [5,4])
22-
ax = fig.add_subplot(111)
22+
fig, ax = plt.subplots(figsize=[5, 4])
2323

2424
Z, extent = get_demo_image()
2525

@@ -28,7 +28,7 @@ def get_demo_image():
2828
ylim=(-20, 5))
2929

3030

31-
axins = zoomed_inset_axes(ax, 2, loc=2) # zoom = 6
31+
axins = zoomed_inset_axes(ax, 2, loc=2) # zoom = 6
3232
im = axins.imshow(Z, extent=extent, interpolation="nearest",
3333
origin="lower")
3434

@@ -38,17 +38,14 @@ def get_demo_image():
3838

3939
# colorbar
4040
cax = inset_axes(axins,
41-
width="5%", # width = 10% of parent_bbox width
42-
height="100%", # height : 50%
41+
width="5%", # width = 10% of parent_bbox width
42+
height="100%", # height : 50%
4343
loc=3,
4444
bbox_to_anchor=(1.05, 0., 1, 1),
4545
bbox_transform=axins.transAxes,
4646
borderpad=0,
4747
)
4848

49+
colorbar(im, cax=cax)
4950

50-
colorbar(im, cax=cax) #, ticks=[1,2,3])
51-
52-
53-
plt.draw()
5451
plt.show()

‎examples/axes_grid1/demo_colorbar_with_axes_divider.py‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -14,14 +14,14 @@
1414
fig.subplots_adjust(wspace=0.5)
1515

1616
ax1 = fig.add_subplot(121)
17-
im1 = ax1.imshow([[1,2],[3,4]])
17+
im1 = ax1.imshow([[1, 2], [3, 4]])
1818

1919
ax1_divider = make_axes_locatable(ax1)
2020
cax1 = ax1_divider.append_axes("right", size="7%", pad="2%")
2121
cb1 = colorbar(im1, cax=cax1)
2222

2323
ax2 = fig.add_subplot(122)
24-
im2 = ax2.imshow([[1,2],[3,4]])
24+
im2 = ax2.imshow([[1, 2], [3, 4]])
2525

2626
ax2_divider = make_axes_locatable(ax2)
2727
cax2 = ax2_divider.append_axes("top", size="7%", pad="2%")

‎examples/axes_grid1/demo_fixed_size_axes.py‎

Lines changed: 7 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -8,8 +8,8 @@
88

99
from mpl_toolkits.axes_grid1 import Divider, LocatableAxes, Size
1010

11-
def demo_fixed_size_axes():
1211

12+
def demo_fixed_size_axes():
1313
fig1 = plt.figure(1, (6, 6))
1414

1515
# The first items are for padding and the second items are for the axes.
@@ -25,19 +25,16 @@ def demo_fixed_size_axes():
2525

2626
fig1.add_axes(ax)
2727

28-
ax.plot([1,2,3])
29-
30-
28+
ax.plot([1, 2, 3])
3129

3230

3331
def demo_fixed_pad_axes():
34-
3532
fig = plt.figure(2, (6, 6))
3633

37-
# The first & third items are for padding and the second items are for the axes.
38-
# sizes are in inch.
39-
h = [Size.Fixed(1.0), Size.Scaled(1.), Size.Fixed(.2),]
40-
v = [Size.Fixed(0.7), Size.Scaled(1.), Size.Fixed(.5),]
34+
# The first & third items are for padding and the second items are for the
35+
# axes. Sizes are in inches.
36+
h = [Size.Fixed(1.0), Size.Scaled(1.), Size.Fixed(.2)]
37+
v = [Size.Fixed(0.7), Size.Scaled(1.), Size.Fixed(.5)]
4138

4239
divider = Divider(fig, (0.0, 0.0, 1., 1.), h, v, aspect=False)
4340
# the width and height of the rectangle is ignored.
@@ -47,16 +44,11 @@ def demo_fixed_pad_axes():
4744

4845
fig.add_axes(ax)
4946

50-
ax.plot([1,2,3])
51-
52-
53-
54-
47+
ax.plot([1, 2, 3])
5548

5649

5750
if __name__ == "__main__":
5851
demo_fixed_size_axes()
5952
demo_fixed_pad_axes()
6053

61-
plt.draw()
6254
plt.show()

‎examples/axes_grid1/demo_new_colorbar.py‎

Lines changed: 8 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -5,21 +5,20 @@
55
66
"""
77
import matplotlib.pyplot as plt
8+
from mpl_toolkits.axes_grid1.colorbar import colorbar
9+
810

9-
plt.rcParams["text.usetex"]=False
11+
plt.rcParams["text.usetex"] = False
1012

11-
fig = plt.figure(1, figsize=(6, 3))
13+
fig, (ax1, ax2) = plt.subplots(1, 2, figsize=(6, 3))
1214

13-
ax1 = fig.add_subplot(121)
14-
im1 = ax1.imshow([[1,2],[3,4]])
15-
cb1 = plt.colorbar(im1)
15+
im1 = ax1.imshow([[1, 2], [3, 4]])
16+
cb1 = fig.colorbar(im1, ax=ax1)
1617
cb1.ax.set_yticks([1, 3])
1718
ax1.set_title("Original MPL's colorbar w/\nset_yticks([1,3])", size=10)
1819

19-
from mpl_toolkits.axes_grid1.colorbar import colorbar
20-
ax2 = fig.add_subplot(122)
21-
im2 = ax2.imshow([[1,2],[3,4]])
22-
cb2 = colorbar(im2)
20+
im2 = ax2.imshow([[1, 2], [3, 4]])
21+
cb2 = colorbar(im2, ax=ax2)
2322
cb2.ax.set_yticks([1, 3])
2423
ax2.set_title("AxesGrid's colorbar w/\nset_yticks([1,3])", size=10)
2524

‎examples/axes_grid1/parasite_simple.py‎

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -27,4 +27,3 @@
2727
leg.texts[1].set_color(p2.get_color())
2828

2929
plt.show()
30-

‎examples/axes_grid1/simple_axes_divider1.py‎

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -30,6 +30,4 @@
3030
ax3.set_axes_locator(divider.new_locator(nx=2, ny=2))
3131
ax4.set_axes_locator(divider.new_locator(nx=2, nx1=4, ny=0))
3232

33-
34-
plt.draw()
3533
plt.show()

‎examples/axes_grid1/simple_axes_divider2.py‎

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@
1212

1313
# the rect parameter will be ignore as we will set axes_locator
1414
rect = (0.1, 0.1, 0.8, 0.8)
15-
ax = [fig1.add_axes(rect, label="%d"%i) for i in range(4)]
15+
ax = [fig1.add_axes(rect, label="%d" % i) for i in range(4)]
1616

1717
horiz = [Size.Scaled(1.5), Size.Fixed(.5), Size.Scaled(1.),
1818
Size.Scaled(.5)]
@@ -31,5 +31,4 @@
3131
plt.setp(ax1.get_xticklabels()+ax1.get_yticklabels(),
3232
visible=False)
3333

34-
plt.draw()
3534
plt.show()

‎examples/axes_grid1/simple_axes_divider3.py‎

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@
1313

1414
# the rect parameter will be ignore as we will set axes_locator
1515
rect = (0.1, 0.1, 0.8, 0.8)
16-
ax = [fig1.add_axes(rect, label="%d"%i) for i in range(4)]
16+
ax = [fig1.add_axes(rect, label="%d" % i) for i in range(4)]
1717

1818

1919
horiz = [Size.AxesX(ax[0]), Size.Fixed(.5), Size.AxesX(ax[1])]
@@ -40,5 +40,4 @@
4040
plt.setp(ax1.get_xticklabels()+ax1.get_yticklabels(),
4141
visible=False)
4242

43-
plt.draw()
4443
plt.show()

‎examples/axes_grid1/simple_colorbar.py‎

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -9,12 +9,11 @@
99
import numpy as np
1010

1111
ax = plt.subplot(111)
12-
im = ax.imshow(np.arange(100).reshape((10,10)))
12+
im = ax.imshow(np.arange(100).reshape((10, 10)))
1313

1414
# create an axes on the right side of ax. The width of cax will be 5%
1515
# of ax and the padding between cax and ax will be fixed at 0.05 inch.
1616
divider = make_axes_locatable(ax)
1717
cax = divider.append_axes("right", size="5%", pad=0.05)
1818

1919
plt.colorbar(im, cax=cax)
20-

‎examples/axes_grid1/simple_rgb.py‎

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -8,23 +8,25 @@
88

99
from mpl_toolkits.axes_grid1.axes_rgb import RGBAxes
1010

11+
1112
def get_demo_image():
1213
import numpy as np
1314
from matplotlib.cbook import get_sample_data
1415
f = get_sample_data("axes_grid/bivariate_normal.npy", asfileobj=False)
1516
z = np.load(f)
1617
# z is a numpy array of 15x15
17-
return z, (-3,4,-4,3)
18+
return z, (-3, 4, -4, 3)
19+
1820

1921
def get_rgb():
2022
Z, extent = get_demo_image()
2123

22-
Z[Z<0] = 0.
23-
Z = Z/Z.max()
24+
Z[Z < 0] = 0.
25+
Z = Z / Z.max()
2426

25-
R = Z[:13,:13]
26-
G = Z[2:,2:]
27-
B = Z[:13,2:]
27+
R = Z[:13, :13]
28+
G = Z[2:, 2:]
29+
B = Z[:13, 2:]
2830

2931
return R, G, B
3032

@@ -39,6 +41,4 @@ def get_rgb():
3941
ax.RGB.set_xlim(0., 9.5)
4042
ax.RGB.set_ylim(0.9, 10.6)
4143

42-
43-
plt.draw()
4444
plt.show()

0 commit comments

Comments
 (0)