import os
import csv

# 文件路径
output_file = r'D:\计量经济学\计量实验资料及作业要求\计量实验资料及作业要求\UGC回归数据.csv'

print("========================================")
print("  创建并填充UGC回归数据")
print("========================================")
print(f"输出文件: {output_file}")
print()

# 检查输出目录是否存在
output_dir = os.path.dirname(output_file)
print(f"输出目录: {output_dir}")
print(f"目录存在: {os.path.exists(output_dir)}")

if not os.path.exists(output_dir):
    print("正在创建输出目录...")
    try:
        os.makedirs(output_dir)
        print("目录创建成功")
    except Exception as e:
        print(f"创建目录失败: {e}")
        exit(1)

# 创建并填充CSV文件
try:
    print("\n创建并填充CSV文件...")
    with open(output_file, 'w', newline='', encoding='utf-8-sig') as f:
        writer = csv.writer(f)
        
        # 写入表头
        headers = ['Y', 'X1', 'X2', 'X3', 'X4', 'X5', 'X6']
        writer.writerow(headers)
        
        # 写入示例数据（前10行）
        for i in range(1, 11):
            row = [
                i * 0.5,      # Y: UGC有用性
                i * 2,        # X1: 评论数量
                i * 10,       # X2: 评论长度
                i * 2,        # X3: 评论复杂度
                5.0,          # X4: 评论可读性
                (i % 3) - 1,  # X5: 内容情感性
                i * 0.3       # X6: 信息丰富度
            ]
            writer.writerow(row)
    
    print(f"文件已成功创建: {output_file}")
    print(f"文件大小: {os.path.getsize(output_file) / 1024:.2f} KB")
    
    # 读取并显示文件内容
    print("\n文件内容:")
    with open(output_file, 'r', encoding='utf-8-sig') as f:
        reader = csv.reader(f)
        for i, row in enumerate(reader):
            if i < 5:
                print(f"行 {i+1}: {row}")
    
    print()
    print("========================================")
    print("  任务完成")
    print("========================================")
    
except Exception as e:
    print(f"处理文件时出错: {str(e)}")
    import traceback
    traceback.print_exc()