执行代码时出现以下错误:
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))
答案 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])
没有意义。 i
是arr
的元素,而不是索引。
这应该有效:
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的位置。