Skip to content

Instantly share code, notes, and snippets.

@wbhinton
Last active January 13, 2020 18:25
Show Gist options
  • Save wbhinton/6de7b424a407bcf4627b4403a9a926a1 to your computer and use it in GitHub Desktop.
Save wbhinton/6de7b424a407bcf4627b4403a9a926a1 to your computer and use it in GitHub Desktop.
Format a string for SQL IN statement from Pandas Dataframe Column

The following snippet generates a regex string that can be used to query DF's

r = "'("
for i in df['columnA']:
    r = r+str(i)+"|"
r = r[:-1]+")'"

The following formats a Pandas DF column into a list for an SQL IN statement

s ='('
for i in df['columnA']:
    s = s+"'"+str(i)+"',"
s = s[:-1]+')'

Which would be used as follows:

sql = """
SELECT 
      COL1
      ,COL2
  FROM TABLE1
  WHERE COL1 in 
"""
sql = sql + s

df_new = pd.read_sql(sql, cnxn)

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment