浏览代码

word_set.remove('') ,之前需要判断是否存在该key

liyuyi@c-top.com.cn 3 年之前
父节点
当前提交
d0a6162725
共有 1 个文件被更改,包括 2 次插入1 次删除
  1. 2 1
      time_task/get_material_and_script_by_query_word.py

+ 2 - 1
time_task/get_material_and_script_by_query_word.py

@@ -38,7 +38,8 @@ def get_material_and_script(start_time=""):
     query_word_set = set(df['query_word'].values)
     query_word_set = set(df['query_word'].values)
     recommended_word_set = set(df['recommended_word'].values)
     recommended_word_set = set(df['recommended_word'].values)
     word_set = query_word_set.union(recommended_word_set)
     word_set = query_word_set.union(recommended_word_set)
-    word_set.remove('')
+    if '' in word_set:
+        word_set.remove('')
     query_word_time = 3
     query_word_time = 3
     for query_word in word_set:
     for query_word in word_set:
         for source_code, value in source_name_map.items():
         for source_code, value in source_name_map.items():