修改测试文件

732ef74b · 张彦钊 · 0f0fc498 · 732ef74b
Commit 732ef74b authored Apr 25, 2019 by 张彦钊
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

multi.py tensnsorflow/multi.py +1 -1

No files found.
--- a/tensnsorflow/multi.py
+++ b/tensnsorflow/multi.py
@@ -80,7 +80,7 @@ def feature_engineer():
                              "channel", "top", "time", "stat_date", "app_list", "hospital_id", "level3_ids"])

    df = df.fillna("na")
-    v = df.select("app_list").collect()
+    v = df.select("app_list").map(lambda x:x[0]).collect()
    print(type(v))
    print(v[:2])