27 lines
723 B
Python
27 lines
723 B
Python
|
import requests
|
|||
|
|
import zipfile
|
|||
|
|
import os
|
|||
|
|
|
|||
|
|
# 下载Telco Customer Churn数据集
|
|||
|
|
def download_telco_churn():
|
|||
|
|
# 使用公开可访问的数据集URL
|
|||
|
|
url = "https://raw.githubusercontent.com/IBM/telco-customer-churn-on-icp4d/master/data/Telco-Customer-Churn.csv"
|
|||
|
|
|
|||
|
|
# 创建data目录（如果不存在）
|
|||
|
|
os.makedirs("data", exist_ok=True)
|
|||
|
|
|
|||
|
|
# 下载文件
|
|||
|
|
response = requests.get(url)
|
|||
|
|
response.raise_for_status()
|
|||
|
|
|
|||
|
|
# 保存文件
|
|||
|
|
file_path = "data/Telco-Customer-Churn.csv"
|
|||
|
|
with open(file_path, "wb") as f:
|
|||
|
|
f.write(response.content)
|
|||
|
|
|
|||
|
|
print(f"数据集已成功下载到 {file_path}")
|
|||
|
|
return file_path
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
download_telco_churn()
|