请给下面代码添加注释 import numpy as npdef least_squares(X, y): X = np.array(X) y = np.array(y) w = np.linalg.inv(X.T @ X) @ X.T @ y return wX = np.array([[1, 2], [1, 3], [1, 4]])y = np.array([2, 3, 4])w = least_squares(X, y)print(w)

好的，我来为代码添加注释。

# 导入 numpy 库并使用别名 np
import numpy as np

# 定义最小二乘法函数
def least_squares(X, y):
    # 将 X 和 y 转换成 numpy 数组
    X = np.array(X)
    y = np.array(y)
    
    # 使用矩阵运算计算最小二乘法系数
    w = np.linalg.inv(X.T @ X) @ X.T @ y
    
    # 返回最小二乘法系数
    return w

# 定义输入数据
X = np.array([[1, 2], [1, 3], [1, 4]])
y = np.array([2, 3, 4])

# 调用最小二乘法函数并获取系数
w = least_squares(X, y)

# 打印系数
print(w)

希望这些注释能帮助理解代码。