python 提取字符_python字符串截取指定内容

激活谷笔记 • 2025-04-16 15:21 • 阅读 117

在Python中提取数据库表的字段名可以通过以下几种方法：

使用数据库连接和游标

对于PostgreSQL数据库，可以使用`psycopg2`库：

 import psycopg2 conn = psycopg2.connect（user, host, port, database, password） cur = conn.cursor（） cur.execute（'SELECT column_name FROM information_schema.columns WHERE table_name = %s', （table_name,）） columns = [row for row in cur.fetchall（）] for column in columns: print（column） cur.close（） conn.close（）

对于MySQL数据库，可以使用`pymysql`库：

 import pymysql conn = pymysql.connect（host='127.0.0.1', user='root', password='', db='study', autocommit=True） cur = conn.cursor（） sql = 'SELECT COLUMN_NAME FROM information_schema.COLUMNS WHERE table_name = %s' cur.execute（sql, （table_name,）） for field in cur.fetchall（）: print（field） cur.close（） conn.close（）

对于SQLite数据库，可以使用`sqlite3`库：

 import sqlite3 conn = sqlite3.connect（'mydb.db'） cur = conn.cursor（） cur.execute（'PRAGMA table_info（your_table_name）'） columns = [description for description in cur.fetchall（）] for column in columns: print（column） cur.close（） conn.close（）

使用PyHive库（针对Hive数据库）：

 from pyhive import hive conn = hive.Connection（host='your_hive_host', port=10000, username='your_username'） cursor = conn.cursor（） cursor.execute（'SHOW COLUMNS FROM your_table_name'） columns = [row for row in cursor.fetchall（）] for column in columns: print（column） cursor.close（） conn.close（）

请根据您使用的数据库类型选择合适的方法，并替换示例代码中的`your_table_name`、`your_username`、`your_hive_host`、`your_password`等参数为您实际的数据库信息。

python 提取字符_python字符串截取指定内容

使用数据库连接和游标

相关推荐