Несколько фигур с подзаговорами с использованием комбинированного списка в Matplotlib

Question

Я хочу создать стримплот из векторного поля, который содержит несколько свободных констант, которые я хотел бы изменить. Итак, я создал комбинации этих констант, и я могу успешно построить графики потоков по одному с помощью этого:

Y, X = np.mgrid[-1:10:200j, 0:10:200j]

tau_x = [0.01, 0.1, 1., 10.]
tau_y = [0.01, 0.1, 1., 10.]
alpha = [0.01, 0.1, 1., 10.]
r = [0.1, 0.01, 0.001]
K = [0.1, 0.5, 1.0, 1.5]

combinations_list = list(itertools.product(tau_x,tau_y,alpha,r,K))

for a in combinations_list:

    (tau_x, tau_y, alpha, r, K) = a

    Fx = (1/tau_x) * ( (-8/3)*(2*r-alpha)*(X-1) + K*X )
    Fy = (2/(tau_y*X**(3/2))) * ( -2*(Y-1) + 3*Y*(X-1)/X + K*X*Y )

    fig, ax = plt.subplots()
    strm = ax.streamplot(X, Y, Fx, Fy, linewidth=0.5)

    plt.show()

Теперь, поскольку мы говорим об очень большом количестве комбинаций, я хотел бы создать фигуру с подзадачами (скажем, 9 на каждую фигуру, но может быть и больше), которая значительно сократила бы количество фигур.

Примечание: мне интересно видеть одну фигуру каждый раз, и поэтому plt.show () находится внутри цикла, чтобы не открывать сразу все фигуры.

РЕДАКТИРОВАТЬ: После sugestion ImportanceOfBeingErnest я изменил код на

Y, X = np.mgrid[-1:10:200j, 0:10:200j]

tau_x = [0.01, 0.1, 1., 10.]
tau_y = [0.01, 0.1, 1., 10.]
alpha = [0.01, 0.1, 1., 10.]
r = [0.1, 0.01, 0.001]
K = [0.1, 0.5, 1.0, 1.5]

combinations_list = list(itertools.product(tau_x,tau_y,alpha,r,K))
length = len(combinations_list)

N = 9 #number of subplots per figure

for i in range(0,100):

    subset = combinations_list[9*i:9*i+9]

    fig = plt.figure()

    j = 1
    for a in subset:

        (tau_x, tau_y, alpha, r, K) = a

        Fx = (1/tau_x) * ( (-8/3)*(2*r-alpha)*(X-1) + K*X )
        Fy = (2/(tau_y*X**(3/2))) * ( -2*(Y-1) + 3*Y*(X-1)/X + K*X*Y )

        ax = fig.add_subplot(3,3,j)
        ax.streamplot(X, Y, Fx, Fy, linewidth=0.5)
        ++j


    plt.show()

но он только чертит первый из каждого подмножества странным образом с цветами в векторах.

Answer 1

Вы не обновляете j правильно.++j не обновляет значение j.Ваш код будет работать нормально, если вы замените ++j на j += 1 или j = j+1.Оба эквивалентны.

for i in range(0,100):
    subset = combinations_list[9*i:9*i+9]
    fig = plt.figure()

    j = 1
    for a in subset:
        (tau_x, tau_y, alpha, r, K) = a

        Fx = (1/tau_x) * ( (-8/3)*(2*r-alpha)*(X-1) + K*X )
        Fy = (2/(tau_y*X**(3/2))) * ( -2*(Y-1) + 3*Y*(X-1)/X + K*X*Y )

        ax = fig.add_subplot(3,3,j)
        ax.streamplot(X, Y, Fx, Fy, linewidth=0.5)
        j += 1 # <--- change here