如何在 Python 3.7 中应用 argrelextrema 函数？

Question

我正在尝试将 argrelextrema 函数与数据帧 df 一起应用。但无法正确申请。下面是我的代码

    import pandas as pd
    from scipy.signal import argrelextrema 
    np.random.seed(42)

        def maxloc(data):
            loc_opt_ind = argrelextrema(df.values, np.greater)
            loc_max = np.zeros(len(data))
            loc_max[loc_opt_ind] = 1
            data['loc_max'] = loc_max
            return data

        values = np.random.rand(23000)
        df = pd.DataFrame({'value': values})
        np.all(maxloc_faster(df).loc_max)

It gives me error 
that loc_max[loc_opt_ind] = 1
IndexError: too many indices for array 

Answer 1

Pandas 数据框是二维的。也就是说，df.values 是二维的，even 当它只有一列时。因此，loc_opt_ind 将包含 x 和 y 索引（两个元组；只需打印 loc_opt_ind 即可查看），不能用于索引 loc_max。您可能想使用df['values'].values（变成<Series>.values）或np.squeeze(df.values) 作为输入。请注意，argrelextrema 在这种情况下仍然返回一个元组，只是一个单元素，因此您可能需要 loc_opt_ind[0]（np.where 具有类似的行为）。