Compartir a través de

Uso de ai.generate_response con pandas

La función ai.generate_response emplea IA generativa para generar, con una sola línea de código, respuestas de texto personalizadas basadas en sus propias instrucciones.

Nota:

Información general

La ai.generate_response función puede extender la clase DataFrame pandas y la clase pandas Series .

Para generar respuestas de texto personalizadas fila por fila, puede llamar a esta función en una serie de Pandas o en un DataFrame de Pandas completo.

Si llama a la función en un DataFrame de pandas completo, el mensaje de entrada puede ser una cadena literal y la función considera todas las columnas del DataFrame al generar las respuestas. El mensaje también puede ser una cadena de formato, donde la función solo tiene en cuenta los valores de columna que aparecen entre llaves en el símbolo del sistema.

La función devuelve una serie pandas que contiene respuestas de texto personalizadas para cada fila de entrada. Las respuestas de texto se pueden almacenar en una nueva columna DataFrame.

Sugerencia

Obtenga información sobre cómo crear mensajes más eficaces para obtener respuestas de mayor calidad siguiendo las sugerencias de solicitud de OpenAI para gpt-4.1.

Syntax

df["response"] = df.ai.generate_response(prompt="Instructions for a custom response based on all column values")

Parámetros

Nombre Description
prompt
Obligatorio		 Cadena que contiene instrucciones de solicitud para aplicar a los valores de texto de entrada de las respuestas personalizadas.
is_prompt_template
Opcional		 Valor booleano que indica si el símbolo del sistema es una cadena de formato o una cadena literal. Si este parámetro se establece Trueen , la función solo tiene en cuenta los valores de fila específicos de cada nombre de columna que aparece en la cadena de formato. En este caso, esos nombres de columna deben aparecer entre llaves y otras columnas se omiten. Si este parámetro se establece en su valor predeterminado de , la función considera todos los valores de Falsecolumna como contexto para cada fila de entrada.
response_format
Opcional		 Diccionario que especifica la estructura esperada de la respuesta del modelo. El type campo se puede establecer en "texto" para texto de forma libre, "json_object" para asegurarse de que la salida es un objeto JSON válido o un esquema JSON personalizado para aplicar una estructura de respuesta específica. Si no se proporciona este parámetro, la respuesta se devuelve como texto sin formato.

Devoluciones

La función devuelve un DataFrame de Pandas que contiene respuestas de texto personalizadas al mensaje de cada fila de texto de entrada.

Example

# This code uses AI. Always review output for mistakes.

df = pd.DataFrame([
        ("Scarves"),
        ("Snow pants"),
        ("Ski goggles")
    ], columns=["product"])

df["response"] = df.ai.generate_response("Write a short, punchy email subject line for a winter sale.")
display(df)

Esta celda de código de ejemplo proporciona la siguiente salida:

Captura de pantalla que muestra una trama de datos con columnas

Ejemplo de formato de respuesta

En el ejemplo siguiente se muestra cómo usar el response_format parámetro para especificar diferentes formatos de respuesta, como texto sin formato, un objeto JSON y un esquema JSON personalizado.

# This code uses AI. Always review output for mistakes.

df = pd.DataFrame([
        ("Alex Rivera is a 24-year-old soccer midfielder from Barcelona who scored 12 goals last season."),
        ("Jordan Smith, a 29-year-old basketball guard from Chicago, averaged 22 points per game."),
        ("William O'Connor is a 22-year-old tennis player from Dublin who won 3 ATP titles this year.")
    ], columns=["bio"])

# response_format : text
df["card_text"] = df.ai.generate_response(
    "Create a player card with the player's details and a motivational quote", 
    response_format={"type": "text"}
)

# response_format : json object
df["card_json_object"] = df.ai.generate_response(
    "Create a player card with the player's details and a motivational quote in JSON", 
    response_format={"type": "json_object"} # Requires "json" in the prompt
)

# response_format : specified json schema
df["card_json_schema"] = df.ai.generate_response(
    "Create a player card with the player's details and a motivational quote", 
    response_format={
        "type": "json_schema",
            "json_schema": {
            "name": "player_card_schema",
            "strict": True,
            "schema": {
                "type": "object",
                "properties": {
                    "name": {"type": "string"},
                    "age": {"type": "integer"},
                    "sport": {"type": "string"},
                    "position": {"type": "string"},
                    "hometown": {"type": "string"},
                    "stats": {"type": "string", "description": "Key performance metrics or achievements"},
                    "motivational_quote": {"type": "string"}
                },
                "required": ["name", "age", "sport", "position", "hometown", "stats", "motivational_quote"],
                "additionalProperties": False,
            },
        },
    },
)

display(df)

Esta celda de código de ejemplo proporciona la siguiente salida:

Captura de pantalla que muestra una trama de datos con una columna

Contenido relacionado

  • Last updated on