Skip to content

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

to_csv add index = False

parent eb8c1bf6
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
diaryCandidateSet.py
View file @ 08b2055c
...@@ -22,7 +22,7 @@ def get_allCitiesDiaryTop2000(): ...@@ -22,7 +22,7 @@ def get_allCitiesDiaryTop2000():
allCitiesTop2000 = con_sql(sql) allCitiesTop2000 = con_sql(sql)
allCitiesTop2000 = allCitiesTop2000.rename(columns={0: "city_id", 1: "cid"}) allCitiesTop2000 = allCitiesTop2000.rename(columns={0: "city_id", 1: "cid"})
allCitiesTop2000 = filter_cid(allCitiesTop2000) allCitiesTop2000 = filter_cid(allCitiesTop2000)
allCitiesTop2000.to_csv(DIRECTORY_PATH + "diaryTestSet/allCitiesDiaryTop2000.csv") allCitiesTop2000.to_csv(DIRECTORY_PATH + "diaryTestSet/allCitiesDiaryTop2000.csv",index=False)
print("成功获取全国日记点击量TOP2000") print("成功获取全国日记点击量TOP2000")
return allCitiesTop2000 return allCitiesTop2000
...@@ -31,7 +31,7 @@ def get_cityList(): ...@@ -31,7 +31,7 @@ def get_cityList():
# 获取全国城市列表 # 获取全国城市列表
sql = "select distinct city_id from data_feed_click" sql = "select distinct city_id from data_feed_click"
cityList = con_sql(sql) cityList = con_sql(sql)
cityList.to_csv(DIRECTORY_PATH + "diaryTestSet/cityList.csv") cityList.to_csv(DIRECTORY_PATH + "diaryTestSet/cityList.csv",index=False)
cityList = cityList[0].values.tolist() cityList = cityList[0].values.tolist()
cityList.remove('worldwide') cityList.remove('worldwide')
print("成功获取全国城市列表") print("成功获取全国城市列表")
...@@ -58,7 +58,7 @@ def get_eachCityDiaryTop2000(): ...@@ -58,7 +58,7 @@ def get_eachCityDiaryTop2000():
pass pass
file_name = DIRECTORY_PATH + "diaryTestSet/{0}DiaryTop2000.csv".format(i) file_name = DIRECTORY_PATH + "diaryTestSet/{0}DiaryTop2000.csv".format(i)
data.to_csv(file_name) data.to_csv(file_name,index=False)
if __name__ == "__main__": if __name__ == "__main__":
......
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