如何獲取Python數據框中多列的斜率

Question

我有下面的數據框，其中包含 4 列分數。 我如何找到數據框中每個單獨 ID 的這 4 個分數的斜率？

ID  t1  t2  t3  t4
a   1   2   3   4
b   3   2   1   
c   4   2   1   2
d   2   3   4   5
e   0   2   3   4

我希望將斜率附加到相同的數據框並在計算斜率后顯示以下內容。

ID  Slope
a   1
b   -1
c   -0.7
d   1
e   1.3

Answer 1

你可以為此使用sklearn （或者可能是scipy ）。 例子：

import sklearn

model = sklearn.linear_model.LinearRegression()
def get_coeff(row, model=model):
    # fit a row assuming points are separated by unit length and return the slope.
    
    row = row.copy().dropna()
    model.fit(np.arange(len(row)).reshape(-1,1), row.values.reshape(-1,1))
    slope = model.coef_[0][0]
    return slope


df["slope"] = df.apply(get_coeff, axis=1)

output：

    t1  t2  t3   t4  slope
ID
a    1   2   3  4.0    1.0
b    3   2   1  NaN   -1.0
c    4   2   1  2.0   -0.7
d    2   3   4  5.0    1.0
e    0   2   3  4.0    1.3