【第十三周】Scipy作业

import numpy as np
import scipy.optimize as opt

#suppose that m=20 n=3
A = np.random.randn(20,3)
B = np.random.randn(20,1)
print("A: ",A)
print("B: ",B)

"""
#suppose that m=8 n=1
A = np.array([[5],[31],[13],[49],[20],[18],[16],[33]])
B = np.array([[33],[72],[70],[10],[21],[6],[21],[52]])
"""

def f(x,target,var):
    return sum(target - var@x)

target = B
var = A
ans=opt.leastsq(f,[[1],[1],[1]] ,args=(target,var))
print("Solution x for Ax=b:\n",ans[0])

运行结果：