寻找numpy ndarray的指数

时间:2016-01-18 15:43:12

标签: python numpy

我有一个numpy.ndarray(数据),它有5个元素,每个元素有2042行和2列。第一列包含日期(每隔15分钟),第二列包含温度。我正在编写一个脚本,可以在24小时内找到最高温度。我有一个工作脚本。

x1=0
y1=95
maxTblue=[] 

for i in range(int(len(data[0])/96)+1):
    #collect the max temp for 24-hr period
    maxTblue.append(max(data[0][x1:y1,1]))
    #add 96 to shift to the next 24-hr period
    x1+=96
    y1+=96

现在,我希望能够收集每24小时最高温度发生的日期。我尝试了以下脚本,但它给了我错误的日期。日期应增加24小时,但当前脚本将返回第一个24小时的日期。

maxdateBlue=[]
locblue=[]
x3=0
y3=95
for i in range(int(len(data[0])/96)+1):
    #index for the max temp 
    locblue.append(np.where(data[0][x3:y3,1]==maxTblue[i]))
    #date where the max temp occurs
    maxdateBlue.append(data[0][locblue[i],0])
    x3+=96
    y3+=96

如何正确使用np.where,或者如何使用其他index命令查明最高温度何时出现?

1 个答案:

答案 0 :(得分:2)

要在numpy数组中查找最大值的索引,请使用np.argmax()

相关问题