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

需要科学

#-*- coding:utf-8 -*-import timeimport reimport osimport openaiimport pandas as pdimport xlsxwriteropenai.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.contentreturn answerdef 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['问题'] # 清空每个单词的句子列表 然后储存这一行新的内容sentence_list = []# 死循环获取输出 防止报错while(True):try:answer = ask_gpt(question)# 成功获取回答 # 可以对answer进行后续处理# passsentence_list.append(answer)break; # 退出循环except Exception as e:# 如果有错误 等10秒钟后继续发起请求print(e)time.sleep(10)continueend_time = time.time()# 计算句子生成时间run_time = end_time-start_timeprint(answer)print(r'消耗时间：{}s'.format(run_time))answer_list.append(sentence_list)# 转换为 DataFramedf = pd.DataFrame(answer_list)# 保存到 Excel 文件df.to_excel('文件名', engine='xlsxwriter', index=False, encoding='utf-8')if __name__ == '__main__':main_word()print("结束")