def create_axs(subplot_n, block=False, return_fig=False):
    r = int(np.floor(np.sqrt(subplot_n)))
    c = int((subplot_n / r))
    fig = plt.figure(figsize=plt.figaspect(0.5))
    axs = {}
    for i in range(subplot_n):
        axs[i] = fig.add_subplot(r, c, (i + 1), projection='3d')
    if return_fig:
        return (axs, fig)
    return axs
def create_axs(subplot_n, block=False, return_fig=False):
    r = int(np.floor(np.sqrt(subplot_n)))
    c = int((subplot_n / r))
    fig = plt.figure(figsize=plt.figaspect(0.5))
    axs = {}
    for i in range(subplot_n):
        axs[i] = fig.add_subplot(r, c, (i + 1), projection='3d')
    if return_fig:
        return (axs, fig)
    return axs