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

问题描述 投票:0回答:1

我是tryig,将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 
python python-3.x pandas numpy scipy
1个回答
0
投票
© www.soinside.com 2019 - 2024. All rights reserved.