我有一个函数,它以numpy数组(A)作为输入。根据数学计算,这个数组可以是一个2d的数组,也可以是一个三维的数组。有一个整数m,它可以是任意数,除非数组是2D,否则m的值总是0。因为A既可以是3D的,也可以是2D的,所以我尝试了3D风格的切片。def fun(A): ans = fun2(A[:,:,m]) #The value of m is 0 if A is 2D
当A是2D时,这给了我一个IndexErrorIndexError: too many i
returns the error below
a[to_select] # works without error, but is not accessing what I want 返回的错误为"IndexError:只有整数、切片(:)、省略号(...)、numpy.newaxis (None)和整数或布尔型数组才是有效的索引“ 有没有办法对子维度使用to_select,或者我需要解决这个问题? 谢谢!