如何在Pandas数据框中获取列的实部和虚部？
import pandas as pd
我有一个数据框：
df=pd.DataFrame({'cmplxnumbers':[1+1j,2-2j,3*(1+1j)]})
我需要获取列中数字的虚部。
我这样做：
df.cmplxnumbers.apply(lambda number: number.imag)
结果我得到：
0 1.0
1 -2.0
2 3.0
Name: cmplxnumbers, dtype: float64
正如预期的那样。
有没有更快、更直接的方法，也许不涉及lambda函数？
回答
Pandas DataFrame/Series 建立在 numpy 数组之上，因此它们可以传递给大多数 numpy 函数。
在这种情况下，您可以尝试以下操作，这应该比非矢量化更快.apply：
df['imag'] = np.imag(df.cmplxnumbers)
df['real'] = np.real(df.cmplxnumbers)
输出：
cmplxnumbers imag real
0 1.000000+1.000000j 1.0 1.0
1 2.000000-2.000000j -2.0 2.0
2 3.000000+3.000000j 3.0 3.0
或者你可以这样做agg：
df[['real','imag']] = df.cmplxnumbers.agg([np.real, np.imag])