Как добавить новые сложные значения в мой массив? - PullRequest
Новая
       23

Как добавить новые сложные значения в мой массив?

1 голос
/ 02 апреля 2019

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

Я пытался использовать numpy.complex(), numpy.vectorize() and normal input (x + 1j*y), но это не работает.Я также попытался нарезать ломтиками.

Я проверил эту проблему и другие места, которые они называют ошибкой.Я перепробовал все их решения, но они не работают для меня. 

import numpy as np
N = 400
Nx = 1334
Cn = np.zeros(shape = (N) )

x0 = 50*10**(-9)
L = 200*10**(-9)
x = np.linspace(0, L, Nx)
sig = 10**(-8)
k = 2289670634.4999747


expsig = np.exp(-((1/2)*(x-x0)**2)/(sig**2))
expimg = np.exp(1j*k*(x-x0))
Phi = ((1/(np.pi**(1/4)*np.sqrt(sig)))*expsig*expimg)

Boxfunc = np.zeros(shape = (N, Nx))
for i in range(0, N):
    SINnpi = np.sin(((i*np.pi)/L)*x)
    Boxfunc[i,:] = np.sqrt(2/L)*SINnpi
    Y = Boxfunc[i,:]*Phi

    realY = np.real(Y)
    imagY = np.imag(Y)

    RealCn = np.trapz(realY)*dx
    ImagCn = np.trapz(imagY)*dx

    Cn[i] = RealCn + 1j*ImagCn

Я ожидаю, что на выходе будет Cn как сложный массив, но все, что я получаю, это: TypeError: can't convert complex to float", or "__main__:59: ComplexWarning: Casting complex values to real discards the imaginary part.

1 Ответ

2 голосов
/ 02 апреля 2019

Попробуйте назначить dtype для вашего массива: 

Cn = np.zeros(shape = (N)).astype('complex128')

По умолчанию np.zeros использует float64.

...