仅Python的整数，切片...以及整数或布尔数组为有效索引

Question

import numpy as np
import matplotlib.pyplot as plt
from scipy.integrate import odeint
from scipy import integrate
from scipy.optimize import fmin

Td=np.array([0.5,1,1.5,2,2.2,3,3.5,4,4.5,5])#time
findindex=lambda x:np.where(mt>=x)[0][0]
mindex=map(findindex,Td)

Zm=Td[mindex]

当我尝试运行此代码时，它将引发异常。

发生异常：IndexError 有效的索引只有整数，切片（:，省略号（...），numpy.newaxis（None）和整数或布尔数组。

我是Python新手。 有人可以解决这个问题吗？

Answer 1

您要过滤符合某些条件的索引的IIUC，然后使用这些索引对numpy数组进行切片

mt=3
Td[np.where(Td<mt)] # where values in Td array less than mt get indexes and slice the array