| 1234567891011121314151617181920212223242526272829 |
- from database.dao.mysql_dao import MySqlDao
- from models.rank.data.config import ProductConfig, ImportanceFeaturesMap
- from models.rank.data.utils import sample_data_clear
- dao = MySqlDao()
- city_uuid = "00000000000000000000000011445301"
-
- order_data = dao.load_order_data(city_uuid)
- order_data["sale_qty"] = order_data["sale_qty"].fillna(0)
- print(order_data.columns.to_list())
- order_data = order_data.infer_objects(copy=False)
-
- # 将销售量进行分组求和
- order_data = order_data.groupby(["stat_month", "cust_code", "product_code"], as_index=False)["sale_qty"].sum()
-
- cust_data = dao.load_cust_data(city_uuid)
- cust_data = cust_data[["BB_RETAIL_CUSTOMER_CODE", "BB_RETAIL_CUSTOMER_NAME"]]
- product_data = dao.load_product_data(city_uuid)
- product_data = product_data[ProductConfig.FEATURE_COLUMNS]
- product_data = sample_data_clear(product_data, ProductConfig)
- sale_data = order_data.merge(cust_data, left_on='cust_code', right_on='BB_RETAIL_CUSTOMER_CODE', how="inner")
- sale_data = sale_data.merge(product_data, left_on='product_code', right_on='product_code', how="inner")
- sale_data = sale_data[["cust_code", "BB_RETAIL_CUSTOMER_NAME"] + ProductConfig.FEATURE_COLUMNS + ["sale_qty", "stat_month"]]
- sale_data = sale_data.rename(columns=ImportanceFeaturesMap.PRODUCT_FEATRUES_MAP)
- sale_data.to_csv("./data/sale_month.csv", index=False)
|
