Skip to content

F
ffm-baseline
Commit 15fe35fb authored by 张彦钊's avatar 张彦钊
Browse files
Options

增加apply参数

parent 64b18a0d
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 2 deletions
tensnsorflow/ffm.py
View file @ 15fe35fb
...@@ -155,7 +155,6 @@ def get_data(): ...@@ -155,7 +155,6 @@ def get_data():
df = df.rename(columns={0: "y", 1: "z", 2: "stat_date", 3: "ucity_id",4: "clevel1_id", 5: "ccity_name", df = df.rename(columns={0: "y", 1: "z", 2: "stat_date", 3: "ucity_id",4: "clevel1_id", 5: "ccity_name",
6:"device_type",7:"manufacturer",8:"channel",9:"top",10:"time",11:"cid_id"}) 6:"device_type",7:"manufacturer",8:"channel",9:"top",10:"time",11:"cid_id"})
print("esmm data ok") print("esmm data ok")
print(df.shape)
db = pymysql.connect(host='10.66.157.22', port=4000, user='root', passwd='3SYz54LS9#^9sBvC', db='jerry_prod') db = pymysql.connect(host='10.66.157.22', port=4000, user='root', passwd='3SYz54LS9#^9sBvC', db='jerry_prod')
sql = "select cid from diary_video" sql = "select cid from diary_video"
diary_video = con_sql(db, sql)[0].values.tolist() diary_video = con_sql(db, sql)[0].values.tolist()
...@@ -165,7 +164,7 @@ def get_data(): ...@@ -165,7 +164,7 @@ def get_data():
video_df["video"] = "yes" video_df["video"] = "yes"
other_df = df.loc[~df["cid_id"].isin(diary_video)] other_df = df.loc[~df["cid_id"].isin(diary_video)]
other_df["video"] = "no" other_df["video"] = "no"
df = video_df.append(other_df) df = video_df.append(other_df).sort_index()
print(df.shape) print(df.shape)
df = df.drop("cid_id",axis=1) df = df.drop("cid_id",axis=1)
print(df.head(2)) print(df.head(2))
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment