python - 我可以获取数据框三列的斜率并使用 python 中的值创建一个新列吗？

Question

有谁知道如何计算每行固定 x 轴的三列（y 轴）的斜率？例如，这是我的数据框：

data = {'a':  [1, 2, 3],
        'b': [3, 4, 5],
        'c': [7, 8, 9]}
df = pd.DataFrame(data, columns=['a', 'b', 'c'])

我需要的是创建第四列“d”，其斜率为 1、3 和 7，考虑到 x 轴是，例如，1、2 和 3。这个 1、2、3 x 轴，应该考虑每一行数据。

标签： pythonnumpyscikit-learnscipylinear-regression

Answer 1

试试这个，你可以np.polyfit()用来找到斜率（通过将多项式的次数定义为 1），然后计算d并将其放入字典中，最后将其转换为pandas DataFrame：

import numpy as np
import pandas as pd

def compute_slope(arr):
    return round(np.polyfit([1, 2, 3], arr, 1)[0], 2)

data = {'a':  [1, 2, 3],
        'b': [3, 4, 5],
        'c': [7, 8, 9]}

data['d'] = [compute_slope(list(arr)) for arr in zip(data['a'], data['b'], data['c'])]
df = pd.DataFrame(data, columns=['a', 'b', 'c', 'd'])

[1,2,3]函数中的变量compute_slope是硬编码的，因为你提到你想要这样的 x 值。

输出：

   a  b  c    d
0  1  3  7  3.0
1  2  4  8  3.0
2  3  5  9  3.0