Update README.md

main
p2apfqmgs 4 months ago
parent 83969d63cc
commit 974b6e2725

@ -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 []

Loading…
Cancel
Save