如何使用Python在 SQL 中查找包含特定列的表？

如果您有许多表格，并想找到具有特定列名称的表，请使用Python进行快速而简便的筛选。

使用Python和SQLAlchemy定位具有特定列的表

这里是在SQLAlchemy中使用Python的核心代码片段

from sqlalchemy import create_engine, inspect


def tables_with_column(engine_str, column_name):
    engine = create_engine(engine_str)
    inspector = inspect(engine)
    tables = inspector.get_table_names()
    result = []
    for table in tables:
        if column_name in [col['name'] for col in inspector.get_columns(table)]:
            result.append(table)
    return result

这段代码使用SQLAlchemy来获取与特定引擎字符串中的数据库建立连接。

然后，它使用连接中的“检查者”对象来查看数据库中包含哪些的表格，然后检查每个表格是否包含所需的列名称。

最后，找到具有特定列名称的所有表格后，tables_with_column返回包含所需列的表格的名称列表。

举例

假设我们正在使用SQLite数据库，并且有以下两个表格：

• orders - 包含一个名为order_id的列和一个名为user_id的列
• users - 包含一个名为user_id的列和一个名为user_name的列

我们只想找到具有user_id列的表格，因此我们将user_id作为参数传递给tables_with_column函数：

>>> engine_str = 'sqlite:///example.db'
>>> column_name = 'user_id'
>>> tables_with_column(engine_str, column_name)
['orders', 'users']

因此，我们可以看到，我们找到了两个具有特定列名的表格。

结论

在Python中使用SQLAlchemy可以很自然地查找包含特定列的表格。 这是一个非常简短的代码片段，具有高度的可读性和可扩展性，可用于连接到各种类型的数据库和生成自定义报告。