python - 使用 Numpy 数组，出现错误

我正在尝试编写函数，为 B-S 模型计算 financial greeks。
我从这样的事情开始:

def greeks_vanilla(S, K, r, q, t, T, sigma):
    import numpy as np
    from scipy.stats import norm

    tau = np.linspace(t, T, num = T)
    d1 = ( np.log(S/K) + (r - q + sigma**2 /2)*tau ) / (sigma * np.sqrt(tau))
    d2 = d1 - sigma*np.sqrt(tau)

    delta_call = np.exp(-q*tau) * norm.cdf(d1)
    delta_put = -np.exp(-q*tau) * norm.cdf(-d1)
    ...

    return {'d1':d1, 'd2': d2, 'delta_call': delta_call, 'delta_put': delta_put, ...}

'...' 的意思是，有更多的希腊语正在计算，但这里并不重要。
它工作正常，我有合理的值(value)，漂亮的情节等；然而，我的老师告诉我，他不仅希望看到这些值与时间(x 轴上的 tau)的关系，还希望看到与 S(S - 股票价格，在 x 轴上)的关系。换句话说，我必须计算 tau 和 S 变化的希腊语。

我试过以下:

def greeks_vanilla(S, K, r, q, t, T, sigma):
    import numpy as np
    from scipy.stats import norm

    S = np.linspace(1, S, num = S)

    tau = np.linspace(t, T, num = T)
    d1 = ( np.log(S/K) + (r - q + sigma**2 /2)*tau ) / (sigma * np.sqrt(tau))
    d2 = d1 - sigma*np.sqrt(tau)

    delta_call = np.exp(-q*tau) * norm.cdf(d1)
    delta_put = -np.exp(-q*tau) * norm.cdf(-d1)
    ...

对于这两个版本，我初始化了以下参数并运行(a 变量):

S = 30.0 #Stock price
K = 50.0 #Strike price
r = 0.05 #Risk-free rate
q = 0.01 #also called delta, annual dividend yield
t = 1.0
T = 100.0
sigma = 0.15
a = greeks_vanilla(S, K, r, q, t, T, sigma)

它在第一种情况下工作正常，但是当我想改变 S(第二个函数)时，出现以下错误:

     File "greeks.py", line 11, in greeks_vanilla
    d1 = ( np.log(S/K) + (r - q + sigma**2 /2)*tau ) / (sigma * np.sqrt(tau))
ValueError: operands could not be broadcast together with shapes (30,) (100,)

我用谷歌搜索了一下，看起来 Numpy 库中有一些东西，它是数据类型(数组)。我不是很熟练或经验丰富的程序员(仍在学习)，所以我无法自己弄清楚。
看起来它现在只适用于 S == 100 (S = T)，但这不是理想的解决方案。
我试过了:

S = list(S)

但它只输出:

File "greeks.py", line 11, in greeks_vanilla
    d1 = ( np.log(S/K) + (r - q + sigma**2 /2)*tau ) / (sigma * np.sqrt(tau))
TypeError: unsupported operand type(s) for /: 'list' and 'float'

请帮我找到这个的工作解决方案。我不知道我是否应该尝试循环 S(我尝试过但失败了......)，做一些数据类型技巧，以其他方式计算(如何？)或其他任何东西。

最佳答案

据我所知，您在 np.linspace 中的每个值处评估您的函数，它适用于一维情况。现在您正在尝试让它在 2D 中工作并且仍在使用 np.linspace 。

Numpy 想要在元素方面工作。所以当你给它两个不同大小的 linspace 时，它​​不知道如何将元素从一个 linspace 映射到另一个中的元素。但是如果你考虑一下你想要什么，你应该意识到你想要在所有对组合 (s, t) 评估你的函数，其中 s in S 和 t in tau ，所有这些对都是二维网格上的点。所以你想使用的工具是 np.meshgrid 。

从图形上看，按照您编写它的方式，您试图仅在 #s 处评估您的表达式。

#...................
#...................
#...................
#...................
#...................
####################

这就是 numpy 提示的原因——它说我不确定 tau 中的哪个值对应于 S 的哪个值。我想做的是将 tau 的一个值与所有 S 匹配或将它们配对，但这只会发生在 len(tau)==len(S) 上，如您所见。您真正想要的是在该网格上的所有点评估您的表达式，这就是 np.meshgrid 会让您做的。

Numpy 的 meshgrid 一开始可能很难理解如何使用。如果你给它提供两个向量(例如你的 linspaces )，它将返回两个二维数组，这些数组指定这两个向量跨越的每个网格点的坐标。我想你可能想做这样的事情:

def greeks_vanilla(S, K, r, q, t, T, sigma):
    import numpy as np
    from scipy.stats import norm

    v1 = np.linspace(1, S, num = S)
    v2 = np.linspace(t, T, num = T)

    S, TAU = np.meshgrid(v1, v2)

    d1 = ( np.log(S/K) + (r - q + sigma**2 /2)*TAU ) / (sigma * np.sqrt(TAU))
    d2 = d1 - sigma*np.sqrt(TAU)

    delta_call = np.exp(-q*TAU) * norm.cdf(d1)
    delta_put = -np.exp(-q*TAU) * norm.cdf(-d1)
    ...

这不再引发错误。

关于python - 使用 Numpy 数组，出现错误，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/33089470/