使用数据库连接和游标
对于PostgreSQL数据库,可以使用`psycopg2`库:
```python
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`库:
```python
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`库:
```python
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数据库):
```python
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`等参数为您实际的数据库信息。