| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657 |
- #coding utf-8
- import sys
- import json
- from db_help import RedisHelper
- if __name__=="__main__":
- nowdate = sys.argv[1]
- f1 = open("./data/sorted_hour_data_"+nowdate)
- f2 = open("./data/sorted_data_"+nowdate)
- data_dict = {}
- while True:
- line = f1.readline()
- if not line:
- break
- items = line.strip().split("\t")
- if len(items)<2:
- continue
- kid = items[0]
- #print(items[1])
- item_info = json.loads(items[1])
- data_dict[kid] = item_info
- f1.close()
- f3 = open("./data/merge_score_"+nowdate, 'w')
- info_dict = {}
- while True:
- line = f2.readline()
- if not line:
- break
- items = line.split("\t")
- if len(items)<2:
- continue
- kid = items[0].replace("k_p:", "")
- kid2 = "k_p2:"+kid
- #print(kid)
- d_item_info = json.loads(items[1])
- if kid in data_dict:
- item_info = data_dict[kid]
- #print("h:",item_info)
- #print("d:",d_item_info)
- total_info = []
- for i in range(len(item_info)):
- total_info.append(0.3*item_info[i]+0.7*d_item_info[i])
- total_item_info = json.dumps(total_info)
- f3.write(kid2+"\t"+total_item_info+"\n")
- info_dict[kid2] = total_item_info
- #print("m:",total_item_info)
- else:
- total_info = []
- for i in range(len(d_item_info)):
- total_info.append(0.7*d_item_info[i])
- total_item_info = json.dumps(total_info)
- f3.write(kid2+"\t"+total_item_info+"\n")
- info_dict[kid2] = total_item_info
- print(info_dict)
- redis_helper = RedisHelper()
- redis_helper.update_batch_setnx_key(info_dict, 60*60*24*15)
- f2.close()
|
