Python Matplotlib Subplot Size Matlab Function