完整的 Python 数据分析案例：直播带货销售额预测

目录

1.

案例背景

代码实现

2. 主要的代码难点解析

2.1 数据清洗 - 缺失值处理

2.2 特征工程 - 新特征计算与独热编码

2.3 特征选择

2.4 模型训练与评估

2.5 数据可视化

3. 可能改进的代码

3.1 数据清洗与特征工程改进

3.2 模型改进

3.3 可视化改进

---

1.

案例背景

直播带货作为当下热门的电商营销模式，吸引了众多消费者和商家的关注。准确预测直播带货的销售额，有助于商家合理安排库存、制定营销策略、优化直播内容和形式，从而提高销售效率和利润。本案例将基于直播带货的相关数据，如主播影响力、直播时长、商品价格、促销活动等，构建销售额预测模型。

代码实现

import pandas as pd
import numpy as np
import matplotlib.pyplot as plt
import seaborn as sns
from sklearn.model_selection import train_test_split
from sklearn.svm import SVR
from sklearn.metrics import mean_squared_error, r2_score
from sklearn.preprocessing import MinMaxScaler

# 数据读取
data = pd.read_csv('live_streaming_sales.csv')

# 数据探索性分析
print('数据基本信息：')
data.info()

# 查看数据集行数和列数
rows, columns = data.shape

if rows < 1000:
    # 小数据集（行数少于 1000）查看全量数据信息
    print('数据全部内容信息：')
    print(data.to_csv(sep='\t', na_rep='nan'))
else:
    # 大数据集查看数据前几行信息
    print('数据前几行内容信息：')
    print(data.head().to_csv(sep='\t', na_rep='nan'))

# 数据清洗
# 处理缺失值
data['anchor_followers'] = data['anchor_followers'].fillna(data['anchor_followers'].mean())
data['discount_rate'] = data['discount_rate'].fillna(data['discount_rate'].median())
data = data.dropna(subset=['sales_amount'])

# 特征工程
# 计算单位时长销售额（销售额 / 直播时长）
data['sales_per_hour'] = data['sales_amount'] / data['live_duration']

# 对商品类别进行独热编码
product_category_dummies = pd.get_dummies(data['product_category'], prefix='product_category')
data = pd.concat([data, product_category_dummies], axis=1)

# 特征选择
features = ['anchor_followers', 'discount_rate', 'sales_per_hour', 'product_price'] + list(product_category_dummies.columns)
target = 'sales_amount'
X = data[features]
y = data[target]

# 数据归一化
scaler = MinMaxScaler()
X_scaled = scaler.fit_transform(X)

# 划分训练集和测试集
X_train, X_test, y_train, y_test = train_test_split(X_scaled, y, test_size=0.2, random_state=42)

# 模型训练
model = SVR(kernel='rbf', C=100, gamma=0.1)
model.fit(X_train, y_train)

# 模型预测
y_pred = model.predict(X_test)

# 模型评估
mse = mean_squared_error(y_test, y_pred)
rmse = np.sqrt(mse)
r2 = r2_score(y_test, y_pred)
print(f"均方误差 (MSE): {mse}")
print(f"均方根误差 (RMSE): {rmse}")
print(f"决定系数 (R²): {r2}")

# 数据可视化
# 不同商品类别的平均销售额柱状图
product_category_avg