12345678910111213141516171819202122232425262728293031323334353637383940414243 |
- # 票 3.0 测试
- import json
- import pandas as pd
- import requests
- appId = "38bb3c94f3004d979b283657e4befb69"
- secret = "kKL7HP0DHxcK1LO5wJhHykDp5w5_Vp70"
- default_header = {
- "appId": appId,
- "secret": secret
- }
- url = "http://aimp.btaimp.k7.bigtree.tech/api/aimp-platform/out_api/processInvoke/ai_piao_model_v3"
- # 读取csv文件,首行为标题
- datas = pd.read_csv('/Users/alvin/Downloads/模型测试_全量样本_for田云.csv')
- # datas转json数组
- datas = datas.to_json(orient='records')
- datas = json.loads(datas)
- # datas 取前10条
- # datas = datas[:2]
- length = len(datas)
- print("datas长度" + str(len(datas))) # datas长度
- a = 0
- for data in datas:
- a += 1
- res = requests.post(url, json=data, headers=default_header).json()
- # 获取p和score {'code': 10000200, 'data': {'output': {'p': '0.005806804', 'score': 1000.0, 'level': 'F'}}}
- if res['code'] == 10000200:
- p_ = res['data']['output']['p']
- score_ = res['data']['output']['score']
- data['p'] = p_
- data['score'] = score_
- print(length - a)
- else:
- print("error")
- print(res)
- # 将datas转为DataFrame,写入csv文件,首行为标题
- datas_df = pd.DataFrame(datas)
- # ["risk_buss_no", "p", ["score"] 为要写入的列
- datas_df[["risk_buss_no", "p", "score"]].to_csv('/Users/alvin/Downloads/模型测试_全量样本_for田云_new.csv', index=False)
|