Как сделать фрейм данных, содержащий комплексные числа из фреймов данных с реальными числами в Python?

Question

В python у меня есть датафрейм с 2 столбцами действительных чисел.Я хочу составить комплексное число и использовать эти два столбца как реальные и мнимые части моего нового набора данных.Я пробовал complex(df['wspd'].astype(float),df['wdir'].astype(float)), но все еще получаю эту ошибку:

cannot convert the series to <class 'float'>

Как я могу это сделать?

Answer 1

Функция complex может обрабатывать только скаляры.Вы можете преобразовать второй столбец в мнимое умножение на 1j, а затем сумму:

df['wspd'] + df['wdir'] * 1j

Образец :

df = pd.DataFrame({'wspd':[10.23,2.4,30.6], 'wdir':[2.3,7.8,4]})
df['com'] = df['wspd'] + df['wdir'] * 1j
print (df)
    wspd  wdir           com
0  10.23   2.3  (10.23+2.3j)
1   2.40   7.8    (2.4+7.8j)
2  30.60   4.0     (30.6+4j)