Json-Python-Server/generate_test_data.py
2026-01-29 18:18:32 +08:00

30 lines
731 B
Python

import pandas as pd
import numpy as np
# 设置随机种子
np.random.seed(42)
# 生成 200 天的时间序列
dates = pd.date_range(start='2023-01-01', periods=200, freq='D')
# 构造数据
trend = np.linspace(0, 50, 200)
seasonality = 10 * np.sin(np.linspace(0, 3.14 * 2 * (200/7), 200))
noise = np.random.normal(0, 2, 200)
sales = 100 + trend + seasonality + noise
ad_cost = sales * 0.5 + np.random.normal(0, 5, 200)
temperature = 30 - trend * 0.2 + np.random.normal(0, 3, 200)
# 创建 DataFrame
df = pd.DataFrame({
'date': dates,
'sales': sales,
'ad_cost': ad_cost,
'temperature': temperature
})
# 保存
df.to_csv('complex_test.csv', index=False)
print("✅ 成功生成测试文件: complex_test.csv")