Use pandas series as input parameter

Question

I'm a R user trying to pick up Python. In R, I often used vectors to pass as arguments to SQL query. For example,

ID <- c(1,2,3,4,5)
df <- dbGetQuery(con, paste("select * from table where ID in (", ID, ")")

How can I achieve this in Python? I have a dataframe and would like to use one of its columns as the parameters. So with a dataframe like this,

data = {'ID': [1,2,3,4,5],
        'Value': [10,20,30,40,50]}
df = pd.DataFrame(data)

[Edit] So basically I need a string that would read "Select * from table where ID in (1,2,3,4,5)" except, instead of manually typing "1,2,3,4,5" I want to use parameters.

Which part are you having trouble with - selecting a column? — wwii
– wwii, Commented Sep 29, 2021 at 19:35
Does select columns based on columns names containing a specific string in pandas anser your question? — wwii
– wwii, Commented Sep 29, 2021 at 19:36
See these answers. To convert a pandas Series to list, use df["ID"].tolist() — Erfan
– Erfan, Commented Sep 29, 2021 at 19:37

Ben.T · Accepted Answer · 2021-09-29 20:22:27Z

1

OP are looking for something like

query = f"select * from table where ID in ({','.join(df['ID'].astype(str))})"

For more ways to create this query from list, one can also check this post provided by @Erfan in a comment.

Collectives™ on Stack Overflow

Use pandas series as input parameter

1 Answer 1

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related