Python应用迭代linregress [关闭]

Question

我正在努力获得以下输出。

数据集优先：

我一直在使用以下代码来获取输出，但每次都失败：

import numpy as np
import pandas as pd
from scipy import stats

df=pd.read_excel('I:/Python/Data/Copy.xlsx')

grouped = df.groupby('Test Event')

for test_event, g in grouped:
    print('Test Event: {}'.format(test_event))
df_np=np.array(g)

x=np.array(df_np[:,3],dtype=float)
y=np.array(df_np[:,4],dtype=float)

for i, pair in enumerate(zip(x, y)):

    slope, intercept, r_value, p_value, std_err = stats.linregress(np.delete(x,i),np.delete(y,i))

    print('slope', slope, 'for data without pair', i, ':', pair) 

---

这是需要的：对于每个测试事件，都有一个唯一的 ID。对于每个唯一 ID，都有一个斜率。图片的评论部分提到了获得斜率的方式。

我该如何解决这个问题？

在jupyter notebook中，最终需要的输出如下：

它只为测试事件 111 和 112 的“没有对 9 的数据”提供输出。

我需要“没有第 0 对的数据”、...、“没有第 9 对的数据”作为每个事件的输出。

Answer 1

您需要将代码缩进如下，以确保循环中有循环：

for test_event, g in grouped:
    # loop over the groups
    print('Test Event: {}'.format(test_event))
    df_np=np.array(g)
    x=np.array(df_np[:,3],dtype=float)
    y=np.array(df_np[:,4],dtype=float)
    for i, pair in enumerate(zip(x, y)):
        # loop over the rows within each group
        slope, intercept, r_value, p_value, std_err = stats.linregress(np.delete(x,i),np.delete(y,i))
        print('slope', slope, 'for data without pair', i, ':', pair)

无需转换为数组和 zip 的稍微简单的方法是：

grouped = df.groupby('Test Event')
df["slope"] = np.NaN
for test_event, g in grouped:
    print('Test Event: {}'.format(test_event))
    for i in g.index:
        others = g.loc[g.index != i, ["x-axis", "y-axis"]]
        slope, intercept, r_value, p_value, std_err = stats.linregress(others)
        print ("slope", slope, 'for data without pair', i)
        df.loc[i, "slope"] = slope

这也将斜率添加到数据框。