用groupby计算数据帧中的多个指数曲线方程

import pandas as pd

# 创建一个示例数据帧
data = {'Group': ['A', 'A', 'B', 'B', 'B', 'C', 'C'],
        'X': [1, 2, 3, 4, 5, 6, 7],
        'Y': [2, 4, 6, 8, 10, 12, 14]}
df = pd.DataFrame(data)

# 使用groupby函数按照Group列进行分组，并计算每个分组的指数曲线方程
grouped = df.groupby('Group')

# 定义计算指数曲线方程的函数
def calculate_equation(group):
    x = group['X']
    y = group['Y']
    # 在这里进行指数曲线方程的计算，例如使用numpy库的polyfit函数
    coefficients = np.polyfit(x, y, deg=2)
    equation = np.poly1d(coefficients)
    return equation

# 对每个分组应用计算函数
equations = grouped.apply(calculate_equation)

# 打印每个分组的指数曲线方程
for group, equation in equations.items():
    print(f"Group {group}: {equation}")