Jupyter SQL 單元格中的 Python 變量

Question

我有一個需要與查詢結果集連接的數據框。 該查詢使用數據框中的列來過濾數據庫中的數據。

data_list = list(df['needed_column'])

我想在 Jupyter sql 單元中執行的 sql 查詢中使用該變量。

%%sql
SELECT
    column_1,
    column_2,
    column_3
FROM my_database.my_table
WHERE
    column_1 IN data_list

無論如何，這可以做到嗎？

Answer 1

一種解決方法是將查詢內聯作為變量執行。

data_list = str(list(df['needed_column']).replace('[', '(').replace(']', ')')

query_string = f"""
SELECT
    column_1,
    column_2,
    column_3
FROM my_database.my_table
WHERE
    column_1 IN {data_list}
"""

result_set = %sql $query_string