python - IndexError:索引666超出轴1的大小501

import math
import numpy as np

def ExplicitMethod(S0, K, r, q, T, Sigma, M, N, Option):

    M = int(M)
    N = int(N)
    dt = T / N
    K = float(K)
    Smax = 2 * K
    dS = Smax / N

    FGrid = np.zeros(shape=(N+1, M+1))

    if Option == 'Call':
        FGrid[-1, :] = np.maximum(np.arange(0, M+1) * dS - K, 0)
    elif Option == 'Put':
        FGrid[-1, :] = np.maximum(K - np.arange(0, M+1) * dS, 0)

    A = np.zeros(shape=(M+1, M+1))
    A[0,0] = 1
    A[-1,-1] = 1
    for j in range(1, M-1):
        A[j, (j-1,j, j+1)] = \
        [0.5 * dt * (Sigma ** 2 * j ** 2 - (r - q) * j  ), \
         1 - dt * (Sigma ** 2 * j ** 2 + r), \
         0.5 * dt * (Sigma **2 * j ** 2 + (r - q) * j)]


    for i in range(N):
        Fhat = FGrid [i, ]
        FGrid[i, :] = np.dot(A , Fhat)
        Fhat[0] = 0
        Fhat[-1] = Smax - K * math.exp(-r * (N - i) * dt)

    k = math.floor(S0 / dS)

    V = FGrid[0, k] + (FGrid[0, k] - FGrid[0, k]) / dS * [(S0 - k * dS)]

    print (V)

ExplicitMethod (50, 30, 0.1, 0.05, 2, 0.3, 400, 800, Option = 'Call')

我犯了个错误：
索引器错误跟踪（最近的调用
最后）在（）
37印（五）
38个
--->39明确方法（50、30、0.1、0.05、2、0.3、400、800，选项='调用'）
用明确的方法（s0，k，r，q，t，
西格玛，M，N，选项）
33 K=数学楼层（s0/ds）
34个
--->35伏=FGrid[0，k]+（FGrid[0，k]-FGrid[0，k]）/dS*[（S0-k*dS）]
36个
37印（五）
索引666：索引轴1超出大小为401的轴1
我在编码期权定价显式方法：
请帮忙，谢谢！

最佳答案

您已将FGrid初始化为（801401）。它告诉我们，出于某种原因，k=666

k = math.floor(S0 / dS)
V = FGrid[0, k] + (FGrid[0, k] - FGrid[0, k]) / dS * [(S0 - k * dS)]

您需要改进如何设置k也许数学是错的。至少你需要确保它不会超过400。

In [200]: S0=50; K=30; N=800
In [201]: dS = 2*K/N
In [202]: dS
Out[202]: 0.075
In [203]: S0/dS
Out[203]: 666.6666666666667

这是你的666。