使用API_KEY向chatGPT（gpt-3.5）批量提问

需要科学

#-*- coding:utf-8 -*-

import time
import re
import os
import openai
import pandas as pd
import xlsxwriter

openai.api_key = "你的API_KEY"

def ask_gpt(prompt):
    '''
        参数：
            prompt : 问题
    
        returns :
            answer :回答
    '''
    completion = openai.ChatCompletion.create(
    model="gpt-3.5-turbo",
    messages=[
        {"role": "user", "content": prompt}
    ]
    )
    answer = completion.choices[0].message.content
    return answer

def main_word():

    # 数据文件地址
    data_file_name = '文件名'

    # 读取数据文件
    data_df = pd.read_excel(data_file_name)
    
    # 定义个空列表 用来储存每行的数据
    answer_list = []

    # 使用 iterrows() 遍历数据文件中每一行 获取词语
    for index, row in data_df[:].iterrows():
        # 生成开始时间
        start_time = time.time()

        # 获取你的问题
        question = row['问题']