Python 使用 OpenAI GPT-3.5-turbo 提取数据倒数第二列
要使用 OpenAI 的 GPT-3.5-turbo 模型编写代码,提取数据的倒数第二列和所有行,可以使用 Python 编程语言和合适的库来实现。下面是一个示例代码:
import openai
import pandas as pd
# 设置 OpenAI API 密钥
openai.api_key = 'YOUR_OPENAI_API_KEY'
# 定义要处理的数据
data = '''
1,John,Doe
2,Jane,Smith
3,Bob,Johnson
4,Alice,Williams
'''
# 处理数据的函数
def extract_data(input_data):
# 将数据转换为 pandas DataFrame
df = pd.read_csv(pd.compat.StringIO(input_data), header=None)
# 提取倒数第二列和所有行
extracted_data = df.iloc[:, -2:]
# 返回提取的数据
return extracted_data.to_string(index=False, header=False)
# 调用 GPT-3.5-turbo 进行提取数据
response = openai.Completion.create(
engine="text-davinci-003",
prompt=f"Extract the last two columns and all rows of the data:
{data}
Result:",
max_tokens=100,
n=1,
stop=None,
temperature=0.3
)
# 获取 GPT-3.5-turbo 的回答
extracted_data = response.choices[0].text.strip()
# 输出提取的数据
print(extracted_data)
请确保替换 YOUR_OPENAI_API_KEY 为您的有效 OpenAI API 密钥。此外,您还需要安装 openai 和 pandas 这两个库,可以通过运行 pip install openai pandas 命令进行安装。
这段代码中,我们首先将要处理的数据存储在 data 变量中。然后,我们定义了一个名为 extract_data 的函数,该函数使用 pandas 库将输入数据转换为 DataFrame,并提取倒数第二列和所有行。最后,我们使用 OpenAI API 的 GPT-3.5-turbo 模型来提取数据,并将结果打印出来。
请注意,GPT-3.5-turbo 的工作方式与之前的 GPT-3.5 模型略有不同,因此我们使用 text-davinci-003 引擎来调用 GPT-3.5-turbo。
原文地址: https://www.cveoy.top/t/topic/PZr 著作权归作者所有。请勿转载和采集!