소스 검색

初始化

罗俊辉 1 년 전
부모
커밋
d9ac747ed7
1개의 변경된 파일5개의 추가작업 그리고 4개의 파일을 삭제
  1. 5 4
      applications/functions.py

+ 5 - 4
applications/functions.py

@@ -30,14 +30,15 @@ class ParamProcess(object):
             if title_tags:
                 for i in range(3):
                     try:
-                        features['tag_{}'.format(i + 1)] = title_tags[i]
+                        features['tag{}'.format(i + 1)] = title_tags[i]
                     except:
                         features['tag_{}'.format(i + 1)] = None
             else:
-                features['tag_1'] = None
-                features['tag_2'] = None
-                features['tag_3'] = None
+                features['tag1'] = None
+                features['tag2'] = None
+                features['tag3'] = None
         df = pd.DataFrame([features])
+        print("data_frame", df.columns)
         df.drop('title', axis=1)
         return df