Spaces:
Running
Running
Commit ·
3bfdad2
1
Parent(s): 53298f7
fix poi_id type
Browse files
app.py
CHANGED
|
@@ -20,6 +20,12 @@ sp = pd.read_csv(STAY_POINTS, nrows=200000)
|
|
| 20 |
poi = pd.read_csv(POI_PATH)
|
| 21 |
demo = pd.read_csv(DEMO_PATH)
|
| 22 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 23 |
# parse act_types string like "[ 2 15]" → list
|
| 24 |
def parse_act(x):
|
| 25 |
try:
|
|
|
|
| 20 |
poi = pd.read_csv(POI_PATH)
|
| 21 |
demo = pd.read_csv(DEMO_PATH)
|
| 22 |
|
| 23 |
+
# 加这两行
|
| 24 |
+
sp["poi_id"] = sp["poi_id"].astype(int)
|
| 25 |
+
poi["poi_id"] = poi["poi_id"].astype(int)
|
| 26 |
+
|
| 27 |
+
sp = sp.merge(poi, on="poi_id", how="left")
|
| 28 |
+
|
| 29 |
# parse act_types string like "[ 2 15]" → list
|
| 30 |
def parse_act(x):
|
| 31 |
try:
|