Python错误IndexError:只有整数

时间:2017-05-28 19:20:10

标签: python numpy

执行代码时出现以下错误: if(abs(arr [i] - val)< float(temp)): IndexError:只有整数,切片(:),省略号(...),numpy.newaxis(None)和整数或布尔数组才是有效索引

代码

def closestval(arr, val):
temp = abs(arr[0] - val)
pos = 0
for i in arr:
    if abs(arr[i] - val) < temp:
        temp = abs(arr[i] - val)
        pos = i
return pos;

我正在传递由np.linspace创建的列表,值为2 * np.pi / 2

代码

x = np.linspace(0, 2*np.pi, 50, endpoint=True)

print(closestval(x, 2*np.pi/2))

1 个答案:

答案 0 :(得分:3)

在python中,当你迭代某个东西时,你会从那个东西中获取元素。你没有得到指数(至少不是自动)

In [262]: for i in ['a','b','c']:
     ...:     print(i)
     ...:     
a
b
c

In [264]: for i in np.arange(10,20,2):print(i)
10
12
14
16
18
In [265]: for i in range(4):print(i)
0
1
2
3

有效地,最后一个表达式在列表[0,1,2,3]上迭代。

所以表达式:

for i in arr:
    print(arr[i])

没有意义。 iarr的元素,而不是索引。

这应该有效:

for a in arr:
    if abs(a - val) < temp:
        temp = abs(a - val)
        #pos = i

但是由于你需要索引i,首选的python迭代是:

for i, a in enumerate(arr):
    av = abs(a - val)
    if av < temp:
        temp = av
        pos = i

其中enumerate添加索引。保持这个enumerate方便。

但这是numpy我们不需要迭代(至少在Python中没有显式)

In [266]: x = np.linspace(0, 2*np.pi, 50, endpoint=True)
In [267]: x<(2*np.pi/2)
Out[267]: 
array([ True,  True,  True,  True,  True,  True,  True,  True,  True,
        True,  True,  True,  True,  True,  True,  True,  True,  True,
        True,  True,  True,  True,  True,  True,  True, False, False,
       False, False, False, False, False, False, False, False, False,
       False, False, False, False, False, False, False, False, False,
       False, False, False, False, False], dtype=bool)
In [268]: np.where(x<(2*np.pi/2))
Out[268]: 
(array([ 0,  1,  2,  3,  4,  5,  6,  7,  8,  9, 10, 11, 12, 13, 14, 15, 16,
        17, 18, 19, 20, 21, 22, 23, 24], dtype=int32),)
In [269]: x[24:26]
Out[269]: array([ 3.07747852,  3.20570679])

我们可以将x的每个元素与一个语句的目标进行比较,并找到最大的。

In [272]: np.max(np.where(x<(2*np.pi/2)))
Out[272]: 24

In [273]: np.argmin(x<(2*np.pi/2))
Out[273]: 25

有多种方法可以确定<测试为True的最后一个元素,或者它切换为False的位置。