如何根据 pandas 中的现有列创建新列？

Question

我是 Python/熊猫的新手。 我想根据“GOOG”价格计算连续回报。 如果价格在 (a) 栏中； 我应该如何根据以下公式计算（b）栏中的回报？

持续回报 =

我想在 Pandas DataFrame 中像下图那样执行此操作（在 Excel 中计算连续回报）。

import pandas as pd

x = pd.DataFrame([2340, 2304, 2238, 2260, 2315, 2318, 2300, 2310, 2353, 2350],
                 columns=['a'])

Answer 1

尝试：

x['b'] = np.log(x['a']/x['a'].shift())

Output：

      a         b
0  2340       NaN
1  2304 -0.015504
2  2238 -0.029064
3  2260  0.009782
4  2315  0.024045
5  2318  0.001295
6  2300 -0.007796
7  2310  0.004338
8  2353  0.018444
9  2350 -0.001276

Answer 2

您可以将生成器 function 与.apply一起使用：

import numpy as np
import pandas as pd

x = pd.DataFrame(
    [2340, 2304, 2238, 2260, 2315, 2318, 2300, 2310, 2353, 2350], columns=["a"]
)


def fn():
    old_a = np.nan
    a = yield
    while True:
        new_a = yield np.log(a / old_a)
        a, old_a = new_a, a


s = fn()
next(s)
x["b"] = x["a"].apply(lambda v: s.send(v))
print(x)

印刷：

      a         b
0  2340       NaN
1  2304 -0.015504
2  2238 -0.029064
3  2260  0.009782
4  2315  0.024045
5  2318  0.001295
6  2300 -0.007796
7  2310  0.004338
8  2353  0.018444
9  2350 -0.001276