要在NumPy中查找datetime数组的索引,您可以使用numpy.where()
函数或布尔索引。以下是两种方法的示例:
方法1:使用numpy.where()
函数
import numpy as np
# 创建一个datetime数组
dates = np.array(['2022-01-01', '2022-01-02', '2022-01-03'], dtype='datetime64')
# 查找特定日期的索引
target_date = np.datetime64('2022-01-02')
index = np.where(dates == target_date)[0]
print(index) # 输出:[1]
在上述示例中,我们使用numpy.where()
函数来查找与目标日期匹配的索引。请注意,numpy.where()
函数返回一个元组,我们使用索引[0]
来获取实际的索引值。
方法2:使用布尔索引
import numpy as np
# 创建一个datetime数组
dates = np.array(['2022-01-01', '2022-01-02', '2022-01-03'], dtype='datetime64')
# 查找特定日期的索引
target_date = np.datetime64('2022-01-02')
index = np.where(dates == target_date)[0]
# 使用布尔索引查找特定日期的索引
bool_index = dates == target_date
index = np.where(bool_index)[0]
print(index) # 输出:[1]
在上述示例中,我们首先创建一个布尔索引,将与目标日期匹配的元素设置为True
,其余元素设置为False
。然后,我们使用numpy.where()
函数来查找True
值的索引。
领取专属 10元无门槛券
手把手带您无忧上云