Индекс 2 выходит за пределы

Question

У меня есть следующий код, который касается ограничения параметров. Я получаю эту ошибку при запуске кода:

samples[:, 2] = np.exp(samples[:, 2]) IndexError: index 2 is out of bounds for axis 1 with size 2

Любая помощь, пожалуйста, как мне сделать, чтобы исправить эту ошибку? Я ценю вашу помощь и ваше внимание. Импорт numpy как np import emcee import matplotlib.pyplot as plt from math import * импорт numpy as np из scipy.integrate импорт quad из scipy.integrate import odeint

xx=np.array([0.01,0.012,0.014,0.016])    #or xx=[0.01.......]
yy=np.array([32.95388698,33.87900347,33.84214074,34.11856704])
Cov=[[137,168],[28155,-2217]]     
#Initial points
rc=0.09, c=0.7, H01 = 70, O_m1 = 0.31, z0=0, M=1, O_m = 0.31, H0=70
np.random.seed(123)

def ant(z,O_m,O_D):          # first function   
    return 1/sqrt(((1+z)**2)*(1+O_m*z)-z*(2+z)*O_D)

def new_calculation(n):        
    O_D=1-O_m
    q=quad(ant,0,xx[n],args=(O_m,O_D))[0]     #using the first function in integration
    h=log10((1+xx[n])*q)   
   fn=(yy[n]-M-h)
   return fn

def log_likelihood(theta):    
    M, O_m= theta
    f_list = []
    for i in range(2):  # the value '2' reflects matrix size
                f_list.append(new_calculation(i))
    rdag=[f_list]
    rmat=[[f] for f in f_list]
    mm=np.dot(rdag,Cov)
    zz=np.dot(mm,rmat)
    hh=np.linalg.det(zz)*0.000001
    return hh          #calculation of matrix


   from scipy.optimize import minimize
np.random.seed(42)
nll = lambda *args: -log_likelihood(*args)
initial = np.array([M, O_m1]) + 0.1*np.random.randn(2)
soln = minimize(nll, initial)
M_ml, O_m0_ml = soln.x

def log_prior(theta):
    M, O_m= theta
    if  0.22 < O_m < 0.32 and 0 < M < 12:
        return 0.0
    return -np.inf

def log_probability(theta):
    lp = log_prior(theta)
    if not np.isfinite(lp):
        return -np.inf
    return lp + log_likelihood(theta)

pos = soln.x + 1e-4*np.random.randn(80, 2)
nwalkers, ndim = pos.shape
sampler = emcee.EnsembleSampler(nwalkers, ndim, log_probability)
sampler.run_mcmc(pos, 250);
samples = sampler.chain[:, 50:, :].reshape((-1, ndim))
from IPython.display import display, Math
samples[:, 2] = np.exp(samples[:, 2])         #the error may be resulted from here
m_mcmc, b_mcmc, f_mcmc = map(lambda v: (v[1], v[2]-v[1], v[1]-v[0]),
                         zip(*np.percentile(samples, [16, 50, 84],
                                            axis=0)))
print(m_mcmc, b_mcmc)

Answer 1

Вы получили 2 параметра O_m а также M который вы установили диапазон для них. но ты получил [:,2] Вот. 2 для 3 параметров! мы начинаем с 0, а не 1. это Numpy. тогда вы хотите определить 3 выхода для вашего кода! m_mcmc, b_mcmc, f_mcmcэто должно быть 2 выхода, а не три. Я не уверен, что такое m, b и f. но я знаю, что один из них должен быть удален. Тогда вы получили ответ.