索引 4 超出了轴 1 的范围，尺寸为 4 带有双和的代码

Question

您好，我有以下函数会产生越界错误：

import numpy as np
import pylab as plt
import scipy
import math
import sympy as sy


T = sy.Symbol('T')
rho = sy.Symbol('rho')


g_T   = [1,T,T**2,T*sy.log(T),T**2*sy.log(T)]
g_rho  = [1,rho,rho**2,rho*sy.log(rho),rho**2*sy.log(rho)]

g_T_np = np.asarray(g_T)
g_rho_np = np.asarray(g_rho)


c = np.loadtxt("c_test.txt")


def F(T,rho):
    ret = 0
    for n in xrange(1,5):
        for m in xrange(1,6):
            inner= c[n,m]*g_T_np*g_rho_np
        ret += inner
    return ret

print F(T,rho)

其中.txt文件是这样的：

-0.529586   -0.000208559    -3.36563E-09    2.29441E-05 
2.22722E-06 -0.00014526 -2.48888E-09    1.89488E-05 
-6.26662E-05    0.000421028 6.17407E-09 -5.14488E-05    
0.09977346  -0.000622051    -8.56485E-09    7.49956E-05 
-0.01437627 -9.86754E-05    -1.59808E-09    1.22574E-05

显示的完整错误是：

Traceback (most recent call last):File "EOS_test.py", line 38, in <module> print F(T,rho) File "EOS_test.py", line 31, in F inner=c[n,m]*g_T_np*g_rho_np IndexError: index 4 is out of bounds for axis 1 with size 4

如何解决这个错误？

Answer 1

您的问题在于设置 xranges。

Python 列表（和 np 数组）的索引为 0，因此您不需要索引 [1,2,3,4,5] 和 [1,2,3,4,5 ,6]，而是想要 ~~[0,1,2,3,4] 和 [0,1,2,3,4,5]~~ [0,1,2,3 ] 和 [0,1,2,3,4].

像这样设置 for 循环可以解决您的问题：

for n in xrange(0,4):
    for m in xrange(0,5):

或者，你可以利用xrange默认的起始点，只列出一个参数：

for n in xrange(4):
    for m in xrange(5):

此外，对于 "more pythonic" 解决方案，不要使用嵌套的 for 循环，而是查找如何遍历 ndarray。 docs举个例子：

it = np.nditer(c, flags=['f_index'])
while not it.finished:
    inner= c[it[0]]*g_T_np*g_rho_np
    ret += inner
    it.iternext()

这避免了需要知道您正在导入的数组大小的整个问题，因此更加稳健。

编辑：正如 pdowling 在他的回答中提到的，范围数字应该是 4 和 5。我在 aswer 中留下了 5 和 6，现在已经改变了。

Answer 2

Numpy 使用基于 0 的索引。从它的外观来看，您正在将数组从 1（第 2 个位置）索引到 4（第 5 个位置），这当然超出了您正在使用的数组的范围。第二轴也是如此。

其次，你弄错了坐标轴：

第一个轴(0)是所选行的索引(0到5)
第二个轴索引列，即行内的值（索引 0 到 4）

这应该有效：

def F(T,rho):
    ret = 0
    for n in range(5):
        for m in range(4):
            inner= c[n,m]*g_T_np*g_rho_np
        ret += inner
    return ret

索引 4 超出了轴 1 的范围，尺寸为 4 带有双和的代码

index 4 is out of bounds for axis 1 with size 4 Code with Double Sum

python

out

bounds