Массив параметра в odeint

Question

Массив параметра в odeint

1

Я пытаюсь решить дифференциальное уравнение с odeint. Здесь некоторые постоянные параметры являются фиксированными, а некоторые находятся в списке.

    from scipy.integrate import odeint
    import matplotlib.pyplot as plt
    import numpy as np
    from scipy.interpolate import LinearNDInterpolator


    #equation of motion in the direction of x    ===== ### d^2x/dt^2 = q[Ex + dy/dt * Bz - dz/dt * By]/m 
    #equation of motion in the direction of y    ===== ### d^2y/dt^2 = q[Ey - dx/dt * Bz + dz/dt * Bx]/m 
    #equation of motion in the direction of z    ===== ### d^2z/dt^2 = q[Ez + dx/dt * By - dy/dt * Bx]/m



    m = 9.1 *(10)**(-31)    
    q = 1.6 *(10)**(-19)    



    #Electric field from FEMM
    with open("Elecric_field_x.txt") as f:
        flines = f.readlines()
        yy1 = [float(line.split()[0]) for line in flines]

    with open("Elecric_field_y.txt") as f:
        flines = f.readlines()
        yy2 = [float(line.split()[0]) for line in flines]

    with open("Elecric_field_z.txt") as f:
        flines = f.readlines()
        yy3 = [float(line.split()[0]) for line in flines]



    #Position x,y,z from FEMM
    with open("Electric_position_x.txt") as f:
        flines = f.readlines()
        y4 = [float(line.split()[0]) for line in flines]

    with open("Electric_position_y.txt") as f:
        flines = f.readlines()
        y5 = [float(line.split()[0]) for line in flines]

        with open("Electric_position_z.txt") as f:
            flines = f.readlines()
            y6 = [float(line.split()[0]) for line in flines]

        #data sample from FEMM inside the text file


    yy1         yy2         yy3          y4         y5          y6


    2.677026732329115255e-01 0.000000000000000000e+00 3.908106187718196067e-01 0.000000000000000000e+00 0.000000000000000000e+00 0.000000000000000000e+00
1.639206109489374516e-17 2.677026732329115255e-01 3.908106187718196067e-01 0.000000000000000000e+00 0.000000000000000000e+00 0.000000000000000000e+00
-2.677026732329115255e-01 3.278412218978749032e-17 3.908106187718196067e-01 -0.000000000000000000e+00 0.000000000000000000e+00 0.000000000000000000e+00
4.031888048269389202e+01 0.000000000000000000e+00 -1.452685819581209046e+02 5.000000000000000278e-02 0.000000000000000000e+00 0.000000000000000000e+00
2.468819396416788133e-15 4.031888048269389202e+01 -1.452685819581209046e+02 3.061616997868383172e-18 5.000000000000000278e-02 0.000000000000000000e+00
-4.031888048269389202e+01 4.937638792833576266e-15 -1.452685819581209046e+02 -5.000000000000000278e-02 6.123233995736766344e-18 0.000000000000000000e+00
-2.020413445543617001e+02 -0.000000000000000000e+00 -2.380940300071312777e+03 1.000000000000000056e-01 0.000000000000000000e+00 0.000000000000000000e+00
-1.237146429519632942e-14 -2.020413445543617001e+02 -2.380940300071312777e+03 6.123233995736766344e-18 1.000000000000000056e-01 0.000000000000000000e+00
2.020413445543617001e+02 -2.474292859039265884e-14 -2.380940300071312777e+03 -1.000000000000000056e-01 1.224646799147353269e-17 0.000000000000000000e+00
0.000000000000000000e+00 0.000000000000000000e+00 0.000000000000000000e+00 0.000000000000000000e+00 0.000000000000000000e+00 1.549999999999999989e-01
0.000000000000000000e+00 0.000000000000000000e+00 0.000000000000000000e+00 0.000000000000000000e+00 0.000000000000000000e+00 1.549999999999999989e-01
-0.000000000000000000e+00 0.000000000000000000e+00 0.000000000000000000e+00 -0.000000000000000000e+00 0.000000000000000000e+00 1.549999999999999989e-01
0.000000000000000000e+00 0.000000000000000000e+00 0.000000000000000000e+00 5.000000000000000278e-02 0.000000000000000000e+00 1.549999999999999989e-01
0.000000000000000000e+00 0.000000000000000000e+00 0.000000000000000000e+00 3.061616997868383172e-18 5.000000000000000278e-02 1.549999999999999989e-01
-0.000000000000000000e+00 0.000000000000000000e+00 0.000000000000000000e+00 -5.000000000000000278e-02 6.123233995736766344e-18 1.549999999999999989e-01
0.000000000000000000e+00 0.000000000000000000e+00 0.000000000000000000e+00 1.000000000000000056e-01 0.000000000000000000e+00 1.549999999999999989e-01
0.000000000000000000e+00 0.000000000000000000e+00 0.000000000000000000e+00 6.123233995736766344e-18 1.000000000000000056e-01 1.549999999999999989e-01
-0.000000000000000000e+00 0.000000000000000000e+00 0.000000000000000000e+00 -1.000000000000000056e-01 1.224646799147353269e-17 1.549999999999999989e-01
0.000000000000000000e+00 0.000000000000000000e+00 0.000000000000000000e+00 0.000000000000000000e+00 0.000000000000000000e+00 3.099999999999999978e-01
0.000000000000000000e+00 0.000000000000000000e+00 0.000000000000000000e+00 0.000000000000000000e+00 0.000000000000000000e+00 3.099999999999999978e-01
-0.000000000000000000e+00 0.000000000000000000e+00 0.000000000000000000e+00 -0.000000000000000000e+00 0.000000000000000000e+00 3.099999999999999978e-01
0.000000000000000000e+00 0.000000000000000000e+00 0.000000000000000000e+00 5.000000000000000278e-02 0.000000000000000000e+00 3.099999999999999978e-01
0.000000000000000000e+00 0.000000000000000000e+00 0.000000000000000000e+00 3.061616997868383172e-18 5.000000000000000278e-02 3.099999999999999978e-01
-0.000000000000000000e+00 0.000000000000000000e+00 0.000000000000000000e+00 -5.000000000000000278e-02 6.123233995736766344e-18 3.099999999999999978e-01
0.000000000000000000e+00 0.000000000000000000e+00 0.000000000000000000e+00 1.000000000000000056e-01 0.000000000000000000e+00 3.099999999999999978e-01
0.000000000000000000e+00 0.000000000000000000e+00 0.000000000000000000e+00 6.123233995736766344e-18 1.000000000000000056e-01 3.099999999999999978e-01
-0.000000000000000000e+00 0.000000000000000000e+00 0.000000000000000000e+00 -1.000000000000000056e-01 1.224646799147353269e-17 3.099999999999999978e-01


        #array for electric field components
    Ex1 = np.array(yy1, dtype=object) 
    Ey1 = np.array(yy2, dtype=object)
    Ez1 = np.array(yy3, dtype=object)



    #array for position
    x = np.array(y4, dtype=object) 
    y = np.array(y5, dtype=object)
    z = np.array(y6, dtype=object)




    def fE(x,y,z,yy1,yy2,yy3,y4,y5,y6):
        #array for electric field components
        Ex1 = np.array(yy1, dtype=object) 
        Ey1 = np.array(yy2, dtype=object)
        Ez1 = np.array(yy3, dtype=object)

        #array for position
        x = np.array(y4, dtype=object) 
        y = np.array(y5, dtype=object)
        z = np.array(y6, dtype=object)

        #linear interpolation of electric field
        ex = LinearNDInterpolator((x, y, z), Ex1)
        ey = LinearNDInterpolator((x, y, z), Ey1)
        ez = LinearNDInterpolator((x, y, z), Ez1)

        #array of new point
        x1 = np.linspace(0, 31, 100)
        y1 = np.linspace(0, 10, 100)
        z1 = np.linspace(0, 10, 100)

        #creating array([x1,y1,z1],[x2,y2,z2],....) for new grids
        X = np.dstack((x1,y1,z1))
        points = np.array(X)

        #Electric field at new grids after linear interpolation
        fEx = ex(points)
        fEy = ey(points)
        fEz = ez(points)
        return fEx, fEy, fEz



    fEx, fEy, fEz = fE(x,y,z,yy1,yy2,yy3,y4,y5,y6)

    #Magnetic field 
    Bx = 0.1825 *(10)**(-4)         
    By = 0.00942 *(10)**(-4)        
    Bz = 0.46264 *(10)**(-4)      




    def trajectory(w, t, p):
        ###====Cartesian coordinate system=====#####
        #x = x1
        #x_prime = y1   #dx/dt
        #y = x2
        #y_prime = y2   #dy/dt
        #z = x3
        #z_prime = y3   #dz/dt

        x1, y1, x2, y2, x3, y3 = w
        q, m, fEx, fEy, fEz, Bx, By, Bz = p

        f = [y1, q*(fEx + y2 * Bz - y3 * By) / m, y2, q*(fEy - y1 * Bz + y3 * Bx) / m, y3, q*(fEz + y1 * By - y2 * Bx) / m] #with magnetic field
        return f


    #Initial conditions
    x1 = 0.0
    y1 = 0.0
    x2 = 0.0
    y2 = 0.0
    x3 = 0.006
    y3 = 68999.35

    #time
    t = np.linspace(0*(10)**(-9), 10.0*(10)**(-9), 100)
    p = [q, m, fEx, fEy, fEz, Bx, By, Bz]
    w0 = [x1, y1, x2, y2, x3, y3]



    # Call the ODE solver.
    wsol = odeint(trajectory, w0, t, args=(p,))

    X = wsol[:,0]       #for x
    Y = wsol[:,2]       #for y
    Z = wsol[:,4]       #for z


    fig = plt.figure()
    ax = fig.add_subplot(111)
    ax.plot(t,X,  color= 'b', label=('x'))
    ax.plot(t,Y, color= 'r', label=('y'))
    ax.plot(t,Z, color= 'c', label=('z'))
    ax.set_xlabel('Time(ns)')
    ax.set_ylabel('position(m)')
    plt.show()

Но я получаю следующую ошибку: Traceback (последний вызов был последним): файл "trajectory_cartesian.py", строка 205, в wsol = odeint (trajectory, w0, t, args = (p,)) ValueError: установка элемента массива с последовательность.

ghimire 09 янв. 2019, в 18:26

Источник

0

Вы пытаетесь вычислить движение заряженной частицы в электромагнитном поле. Если массивы fE являются зависящим от времени электрическим полем, вы также пропустили указание времени выборки и использование интерполяции для вычисления поля в данный момент времени. Пожалуйста, подтвердите или исправьте эти наблюдения и адаптируйте свой код так, чтобы ваше сообщение об ошибке действительно было сообщением об ошибке
LutzL 10 янв. 2019, в 12:53

Теги:

python

scipy

odeint

ode

1 ответ

Ещё вопросы

Вы пытаетесь вычислить движение заряженной частицы в электромагнитном поле. Если массивы fE являются зависящим от времени электрическим полем, вы также пропустили указание времени выборки и использование интерполяции для вычисления поля в данный момент времени. Пожалуйста, подтвердите или исправьте эти наблюдения и адаптируйте свой код так, чтобы ваше сообщение об ошибке действительно было сообщением об ошибке

LutzL · Accepted Answer · 2019-01-10T11-32-00.000Z

недавняя версия вопроса - была проблема XY, нужна динамическая оценка E-поля

Вы пытаетесь передать большую интерполированную сетку значений поля E в функцию ODE. Это не то, что вам нужно. И также не возможно, массив параметров не был предназначен для такой цели. _{(Вот почему это была проблема XY, вы хотите решить X, использовать метод Y и получить проблему, а затем попробовать устранить неполадки Y без сообщения X, но оказывается, что метод Y не является хорошим решением, и вы должны использовать какой-то другой метод Z)}

Для функции ODE необходимо значение E-поля в текущих координатах. Просто сделайте интерполятор глобальным объектом и используйте его в функции ODE, согласно документации по функции интерполяции, это должно работать. Используя заданные точки данных, чтобы заполнить углы куба с длиной стороны 4, читая из строки вместо файлов,

griddata = """  597.8291    0.0         172.9540    -2.0   -2.0   -2.0
                561.7756    204.4696    172.9540    -2.0   -2.0   2.0
                457.9636    384.2771    172.9540    -2.0   2.0    -2.0
                298.9145    517.7352    172.9540    -2.0   2.0    2.0
                103.8119    588.7467    172.9540    2.0    -2.0   -2.0
                -103.8119   588.7467    172.9540    2.0    -2.0   2.0
                -298.9145   517.7352    172.9540    2.0    2.0    -2.0
                -457.9636   384.2771    172.9540    2.0    2.0    2.0""";

grid = [ [ float(cc) for cc in line.split()] for line in griddata.split("\n")];
grid = np.asarray(grid);

xyz_grid = grid[:,3:]   #   xyz_grid = np.array([y4, y5, y6]).T
E_grid = grid[:,:3]     #   E_grid = np.array([yy1, yy2, yy3]).T
E_field = LinearNDInterpolator( xyz_grid, E_grid )

def trajectory(w, t):
    x, vx, y, vy, z, vz = w
    Ex, Ey, Ez = E_field([x, y, z])[0] # returns list of arrays
    f = [ vx, q*(Ex + vy * Bz - vz * By) / m, 
          vy, q*(Ey + vz * Bx - vx * Bz) / m, 
          vz, q*(Ez + vx * By - vy * Bx) / m ]
    return f

Обратите внимание, что здесь все константы, используемые в функции, являются глобальными константами, поэтому вызов

wsol = odeint(trajectory, w0, t)

это следует изменить только в том случае, если q и m являются переменными на разных этапах интеграции.

Вам, вероятно, следует изменить масштаб переменных положения и времени, чтобы координаты и скорости, которые видит odeint находились в диапазоне величин 0.1..10. В противном случае допуски (по умолчанию) могут привести к странным изменениям в отдельных компонентах.

старая версия вопроса, неправильно построенный вектор параметров

lsode оболочки odeint пытается odeint ваш список параметров в массив. Ожидается, что этот список представляет собой плоский список чисел. Ваш список содержит другие списки, которые дают неоднородную структуру, которая не вписывается в numpy массива.

fEx вопросом, для чего предназначены списки fEx и т.д., fEx функция ODE использует эти параметры, как если бы они были числами.

from scipy.integrate import odeint
import numpy as np

m = 9.1e-31
q = 1.6e-19

Bx = 0.1825e-4       
By = 0.00942e-4  
Bz = 0.46264e-4   

fEt = [ 0, 2e-9, 4e-9, 6e-9, 8e-9, 10e-9]
fEx = [0.20507215, 0.20658776, 0.20810338, 0.20961899, 0.21113461, 0.21265022]
fEy = [0.17207596, 0.16972669, 0.16737742, 0.16502815, 0.16267888, 0.1603296]
fEz = [ 3.90810619e-01,  3.60677316e-01,  3.30544013e-01,  3.00410711e-01,   2.70277408e-01,  2.40144105e-01 ]

def trajectory(w, t, p):
    q, m = p  # not really necessary, global variables work here fine
    x1, y1, x2, y2, x3, y3 = w
    Ex, Ey, Ez = np.interp(t,fEt, fEx), np.interp(t,fEt, fEy), np.interp(t,fEt, fEz)
    f = [y1, q*(Ex + y2 * Bz - y3 * By) / m, y2, q*(Ey - y1 * Bz + y3 * Bx) / m, y3, q*(Ez + y1 * By - y2 * Bx) / m]
    return f

x1, y1 = 0.0, 0.0
x2, y2 = 0.0, 0.0
x3, y3 = 0.006, 68999.35

#time 
t = np.arange(0, 10, 0.01)*1e-9

p = [q, m]
w0 = [x1, y1, x2, y2, x3, y3]

# Call the ODE solver.
wsol = odeint(trajectory, w0, t, args=(p,))
print wsol

x1, y1, x2, y2, x3, y3 = wsol.T

from mpl_toolkits.mplot3d import Axes3D
from matplotlib import cm

fig=plt.figure()
ax=fig.gca(projection='3d')
ax.plot(x1,x2,x3,'r',label='charged particle trajectory')
ax.set_xlabel('$x_1$')
ax.set_ylabel('$x_2$')
ax.set_zlabel('$x_3$')
ax.legend()
plt.show()

На самом деле, у меня есть электрическое поле в текстовом файле (Ex, Ey, Ez) и положение x, y, z. Я получил электрическое поле от программного обеспечения FEMM. После этого я создал массив и сделал линейную интерполяцию. fEx и т. д. являются данными электрического поля после линейной интерполяции.
Что означает, что E должно быть функцией от x,y,z ? Тогда объем данных, которые вы разместили, будет недостаточным, а также без какого-либо контекста того, где эти значения фактически взяты. Если у вас есть данные FEM, вам понадобится структура или, возможно, библиотека для чтения и интерпретации данных. Для непосредственной задачи можно считать постоянную E , которой почти равны заданные значения.
Я только что загрузил свой полный сценарий. Не могли бы вы проверить это. заранее спасибо
Извините, я новичок в программировании, получаю сообщение об ошибке: Traceback (последний вызов был последним): файл "trajectory_test.py", строка 150, в <module> wsol = odeint (trajectory, w0, t, args = p) odepack.error: дополнительные аргументы должны быть в кортеже
Да, это было неправильное упрощение, которое я не перенес обратно. Старый код в нижней части теперь должен работать, новая версия в верхней части не имеет параметров, если вы не вставите их обратно.
Сожалею! У меня все еще та же ошибка!
Из какого кода? Python 2.7, и полный скрипт ниже создает картинку без сообщения об ошибке. Очевидно, я не могу проверить фрагмент сверху, так как у меня нет данных сетки. Если вы хотите что-то поднять, для доказательства концепции будет достаточно сетки с 8 вершинами одного куба.
Да, вы правы, но я ожидаю вашей помощи по моему новому вопросу!
Я загрузил данные, пожалуйста, проверьте это!
Данных недостаточно, точки сетки не образуют куб вокруг начала координат, интеграция покидает область интерполяции. Я просто переместил точки в углы куба [-2,2]^3 чтобы получить некоторый результат.
Спасибо! Я получаю вывод, но я ожидаю график от 0,006 до 0 в направлении z, но вот график от 0,006 до 0,0066. Это из-за данных или чего-то еще? Если я хочу несколько строк с различными начальными условиями, то как построить? Заранее спасибо!
Если вы можете дать еще несколько условий, вы можете получить такой путь через решатель solve_bvp задач solve_bvp .
Хорошо, я все еще в замешательстве, если у меня есть данные для координат z от 0 до 10. Когда я использовал начальное условие для z = 10. Тогда траектория должна быть от 10 до 0 в координате z (обратное направление). Но я получаю траекторию от 10 до 20 в z. Почему?
У вас есть 6 начальных условий, которые выбирают, какой путь будет выбран. Просто изменение z не говорит о многом. У вас есть какой-то пример решения, которое заставляет вас сказать, что путь должен идти к z=0 ?
Я загрузил данные, здесь координаты z начинаются с 0 до 0.309. Когда я использую начальное условие для z = 0,30, тогда сюжет начинается с 0,30. Я не знаю, что не так? Я ожидаю сюжет с 0,30 до 0. Спасибо!
Каковы другие начальные условия? Если начальная скорость в направлении z велика и положительна, неудивительно, что значение z растет.
все остальные начальные условия точно такие же, которые я упоминал выше. Хорошо, значение z растет, но это должна быть траектория от 0, я прав?
Да, тогда во время 1e-8 вы достигнете z=0.300689185186 , что близко к z0+dt*vz0=x3[0]+t[-1]*y3[0]=0.300689303506 , на пути полностью доминирует начальный скорость. Почему вы ожидаете что-то другое?
ХОРОШО. Я подаю напряжение внутри области от z = 0 до 0,3. Итак, я хочу увидеть траекторию внутри этого региона.