一个区间内的多个最大值

Question

我认为我正在尝试做的事情在理论上应该很简单，但我就是不知道该怎么做。顺便说一句，在我开始之前，我应该先说我只允许自己使用scipy 和numpy。 （所以更重要的是我不允许使用pandas）

我有这个csv list 和列名'Latitude','Longitude','Time','Quantity'（只是一个sn-p，这个列表很大，有很多年），我想做的是每年找到最大的纬度。我很难区分每年最大的元素。这是我用于分隔列的代码

def read_file(x):
    # Reads the file and returns the 4 columns of datapoints as lists.
    data=np.genfromtxt(x, delimiter=',',usecols=range(4),dtype=None,unpack=True,names=True,encoding=None)
    date,time=np.genfromtxt(data['Time'],delimiter='T',dtype=None,unpack=True,encoding=None)
    year,month,day=np.genfromtxt(date,delimiter='-',dtype=None,unpack=True)
    return year,data['Latitude'] 
# I'm just returning the important things for now
# but the plan is to use all of the data for later tasks

计划是创建另一个函数来查找当前元素是否与前一个元素相同：

def max_per_year(x):
   year,lat=read_file(x)
   for i in range(len(year):
      if year[i]==year[i-1]:

这就是我无法弄清楚如何让它通过以前的is 看到的地方，直到它再次不同。

Answer 1

让我们思考一下这段代码在做什么......

def max_per_year(x):
    latdic={}
    year,lat=read_file(x)
    for i in range(len(year)):
        if year[i]==year[i-1]:
            if float(lat[i]) >=float(lat[i-1]):
                latdic[str(year[i])]=lat[i]
        elif year[i]!=year[i-1]:
            latdic[str(year[i])]=lat[i]
    return latdic, max(lat)

特别看一下这个条件：if float(lat[i]) >=float(lat[i-1]) - 它会用前一个 lat 评估每个 lat，如果更大，则替换当前的 max。

因此，如果我有一个数组 [10, 1, 2]，那么当您需要 10 时，此逻辑将返回 2 作为最大数。您需要做的是与 current max 进行比较，而不是与之前的索引进行比较。

此外，您可以完全放弃年份比较，因为它只是键值对中的一个键。

想一想 - 当您位于数组的开头时，[i-1] 的计算结果是什么？

试试这样的：

def max_per_year(x):
    latdic={}
    year,lat=read_file(x)
    for i in range(len(year)):
        # at least one entry exist for year[i]
        if year[i] in latdic:
            # an entry exists, now check if current entry is less than lat[i]
            if latdic[year[i]] < lat[i]:
                latdic[year[i]]=lat[i]
        # no entry for year[i] key
        else:
            latdic[year[i]]=lat[i]
    return latdic