用numpy解决鸡兔同笼问题

import numpy as np
def get_NumberList(a):
result=[]
strList=a.split(',')
for i in range(len(strList)):
result.append(float(strList[i]))
return result

x=input('请输入鸡兔同笼问题的动物总头数,动物总脚数(逗号隔开):')
#a = np.array([get_NumberList(x), get_NumberList(y)])
b = np.array(get_NumberList(x))
a = np.array([[1,1], [2,4]])
#b = np.array([9,8])
x = np.linalg.solve(a, b)
print('笼子里有兔子: '+str(x[0])+' 个')
print('笼子里有鸡: '+str(x[1])+' 个')
现在发现有好多现成的库就可以直接解决问题!

0 条评论

目前还没有评论...