💡 Как избежать ошибок при SQL-запросах, если в списке только одно значение?

Иногда при выполнении SQL-запросов с оператором IN может возникнуть ошибка, если список содержит только одно значение. Столкнулась с этим на работе, и решила проблему хитрым способом: добавляю фиктивное значение в список.

Это позволяет SQL-запросу работать корректно в любом случае!

from pandasql import sqldf

import pandas as pd



# Список user_id

users_lst = [987654321]



# Проверка длины списка

if len(users_lst) == 1:

    # Добавляем фиктивное значение

    users_lst.append(123456789)  



# Преобразование списка в кортеж

users_lst = tuple(users_lst)



# Пример базы данных с двумя колонками: user_id и amount

data = {

    'user_id': [987654321, 154589364, 234567890, 345678901, 456789012],

    'amount': [100, 200, 300, 400, 500]

}

df = pd.DataFrame(data)



# SQL-запрос

query = f"""

    SELECT user_id, amount

    FROM df

    WHERE user_id IN {users_lst}

"""



# Вывод результата

sqldf(query)

📊 В этом примере, даже если список users_lst содержит одно значение, мы добавляем фиктивный user_id, чтобы запрос был корректным. Используем библиотеку pandasql для выполнения SQL-запроса на DataFrame.