@@ -16,7 +16,7 @@ def annotate_axes(ax, text, fontsize=18):
1616 ax .text (x = 0.5 , y = 0.5 , z = 0.5 , s = text ,
1717 va = "center" , ha = "center" , fontsize = fontsize , color = "black" )
1818
19- # (view , (elev, azim, roll))
19+ # (plane , (elev, azim, roll))
2020views = [('XY' , (90 , - 90 , 0 )),
2121 ('XZ' , (0 , - 90 , 0 )),
2222 ('YZ' , (0 , 0 , 0 )),
@@ -29,18 +29,15 @@ def annotate_axes(ax, text, fontsize=18):
2929 ['.' , '.' , '-XY' , '.' ]]
3030fig , axd = plt .subplot_mosaic (layout , subplot_kw = {'projection' : '3d' },
3131 figsize = (12 , 8.5 ))
32- for i in range (len (axd ) - 1 ):
33- plane = views [i ][0 ]
32+ for plane , angles in views :
3433 axd [plane ].set_xlabel ('x' )
3534 axd [plane ].set_ylabel ('y' )
3635 axd [plane ].set_zlabel ('z' )
3736 axd [plane ].set_proj_type ('ortho' )
38- axd [plane ].view_init (elev = views [i ][1 ][0 ],
39- azim = views [i ][1 ][1 ],
40- roll = views [i ][1 ][2 ])
37+ axd [plane ].view_init (elev = angles [0 ], azim = angles [1 ], roll = angles [2 ])
4138 axd [plane ].set_box_aspect (None , zoom = 1.25 )
4239
43- label = f'{ plane } \n { views [ i ][ 1 ] } '
40+ label = f'{ plane } \n { angles } '
4441 annotate_axes (axd [plane ], label , fontsize = 14 )
4542
4643for plane in ('XY' , '-XY' ):
0 commit comments