How do I store the output of a lang chain pandas dataframe agent to a variable?

Question

I am currently using two data frames and I wish to compare two columns across them, and then create an output variable. However, I want to store the output in the form of a pandas dataframe, or atleast in a json output, which I can then read back to python. My approach is currently using a python dataframe agent, if there's an alternative, would appreciate it.

My datasets look like this

Table 1
| ID  | Food items |
| --- | -----------|
| 1   | Apple      |
| 2   | Celery     |
| 3   | Chicken    |


Table 2
| ID  | Categories |
| ----| -----------|
| 1   | Vegetable  |
| 2   |Fruit       |
| 3   |Meat        |

I would like a mapping of Apple to Fruit, Celery to Vegetable etc in a table format.

Can you give an example of the input data and the result you're trying to achieve? — NotAName
– NotAName, Commented Nov 4, 2024 at 5:54
Well, you need to create rules for such mapping, there's no pandas functionality you can use to linguistically match words. — NotAName
– NotAName, Commented Nov 5, 2024 at 0:28
I wanted to use the langchain functionality to be able to linguistically match the words. — Teestaa Saha
– Teestaa Saha, Commented Nov 8, 2024 at 16:18

Alberto Perez · Accepted Answer · 2024-11-08 16:09:00Z

0

With Pandas:

# Assuming your dataframes are named df1 and df2
result = pd.merge(df1, df2, on='ID')

# If you want to just keep the Food items and Categories columns:
result = result[['Food items', 'Categories']]

answered Nov 8, 2024 at 16:09

Alberto Perez

1,0871 gold badge15 silver badges20 bronze badges

Sign up to request clarification or add additional context in comments.

Comments

Teestaa Saha · Accepted Answer · 2024-11-08 16:16:53Z

0

When the output is generated, store it to a variable which will be in the form of a python dictionary of input and output. For example if your output is in the format,

 input:{query}, output :{"Apple":"Fruit"}

Then, save it to a variable

code_response=input:{query}, output :{"Apple":"Fruit"}

From this dictionary, you can extract output as code df = pd.DataFrame(list(data.items()), columns=['Item','Category']) in your standard python dictionary manipulation.

edited Nov 8, 2024 at 16:16

answered Nov 5, 2024 at 3:25

Teestaa Saha

397 bronze badges

1 Comment

Simon Over a year ago

It could be great if you add an example code for anyone wondering how you applied this anwer

Collectives™ on Stack Overflow

How do I store the output of a lang chain pandas dataframe agent to a variable?

2 Answers 2

Comments

1 Comment

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

1 Comment

Your Answer

Sign up or log in

Post as a guest

Related