piao_test.py 1.4 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243
  1. # 票 3.0 测试
  2. import json
  3. import pandas as pd
  4. import requests
  5. appId = "38bb3c94f3004d979b283657e4befb69"
  6. secret = "kKL7HP0DHxcK1LO5wJhHykDp5w5_Vp70"
  7. default_header = {
  8. "appId": appId,
  9. "secret": secret
  10. }
  11. url = "http://aimp.btaimp.k7.bigtree.tech/api/aimp-platform/out_api/processInvoke/ai_piao_model_v3"
  12. # 读取csv文件,首行为标题
  13. datas = pd.read_csv('/Users/alvin/Downloads/模型测试_全量样本_for田云.csv')
  14. # datas转json数组
  15. datas = datas.to_json(orient='records')
  16. datas = json.loads(datas)
  17. # datas 取前10条
  18. # datas = datas[:2]
  19. length = len(datas)
  20. print("datas长度" + str(len(datas))) # datas长度
  21. a = 0
  22. for data in datas:
  23. a += 1
  24. res = requests.post(url, json=data, headers=default_header).json()
  25. # 获取p和score {'code': 10000200, 'data': {'output': {'p': '0.005806804', 'score': 1000.0, 'level': 'F'}}}
  26. if res['code'] == 10000200:
  27. p_ = res['data']['output']['p']
  28. score_ = res['data']['output']['score']
  29. data['p'] = p_
  30. data['score'] = score_
  31. print(length - a)
  32. else:
  33. print("error")
  34. print(res)
  35. # 将datas转为DataFrame,写入csv文件,首行为标题
  36. datas_df = pd.DataFrame(datas)
  37. # ["risk_buss_no", "p", ["score"] 为要写入的列
  38. datas_df[["risk_buss_no", "p", "score"]].to_csv('/Users/alvin/Downloads/模型测试_全量样本_for田云_new.csv', index=False)