import pandas as pd

# 1. 模拟原始数据（实际开发中你会用 pd.read_csv("file.csv")）
data = {
    '用户ID': [101, 102, 103, 104],
    '逾期天数': [5, 12, 30, 2],
    '最后沟通内容': ['电话未接通', '承诺明天还款', '拒绝还款，态度恶劣', '已提醒']
}

df = pd.DataFrame(data)

# 2. 筛选：只找出逾期天数 > 10 天的用户（AI 只处理高风险的）
high_risk_df = df[df['逾期天数'] > 10]

# 3. 处理：给 AI 准备 Prompt
# 我们要把表格的一行变成一句话
def create_prompt(row):
    return f"用户ID{row['用户ID']}逾期{row['逾期天数']}天，备注是：{row['最后沟通内容']}。请给出催收策略建议。"

# 用 apply 函数批量生成
high_risk_df['AI_Prompt'] = high_risk_df.apply(create_prompt, axis=1)

print("--- 准备发送给 AI 的数据清单 ---")
print(high_risk_df[['用户ID', 'AI_Prompt']])