diff --git a/README.md b/README.md index cfab4d4..03ab318 100644 --- a/README.md +++ b/README.md @@ -32,3 +32,29 @@ special 2.通过用户借阅图书推测出用户喜好并推荐图书 3.统计每种图书的借阅数量供下次买书参考 4.是否需要对用户等级进行分类,比如借数超过50本还书时间限制延长一个周 + +from numpy import linalg +import numpy as np +def student(input_data): + ''' + 将输入数据筛选性别为男,再进行线性方程求解 + :param input_data:类型为`list`的输入数据 + :return:类型为`ndarray` + ''' + result=[] + # ********* Begin *********# + try: + a=np.array(input_data) + d=[] + b1 = [int(row[1]) for row in a if row[0]=="男"] + b2 = [int(row[2]) for row in a if row[0]=="男"] + c = [int(row[-1]) for row in a if row[0]=="男"] + b=np.vstack((b1,b2)) + b=b.transpose() + d=np.linalg.solve(b,c) + d=d.reshape(2,1) + # ********* End *********# + return d + except: + return [] +