Este artículo es una ilustración de cómo extraer nombres de columnas de una tabla de PostgreSQL usando psycopg2 y Python.
Mesa usada para demostración:
Ejemplo 1:
Primero, conectamos la base de datos PostgreSQL usando el método psycopg2.connect() , luego creamos un cursor usando el método cursor(), luego usamos el cursor() y luego extraemos el primer elemento usando el corte.
Python3
import psycopg2 conn = psycopg2.connect( database="geeks", user='postgres', password='pass', host='localhost', port= '5432' ) conn.autocommit = True cursor = conn.cursor() sql = '''SELECT * FROM products''' cursor.execute(sql) column_names = [desc[0] for desc in cursor.description] for i in column_names: print(i) conn.commit() conn.close()
Producción:
product_no name price
Ejemplo 2: en el segundo enfoque, ejecutamos el siguiente comando SQL en el método cursor.execute().
«seleccione COLUMN_NAME de information_schema.columns donde table_schema = ‘SCHEMA_NAME’ y table_name=’TABLE_NAME'»
Python3
import psycopg2 conn = psycopg2.connect( database="geeks", user='postgres', password='root', host='localhost', port='5432' ) conn.autocommit = True with conn: with conn.cursor() as cursor: cursor.execute( "select COLUMN_NAME from information_schema.columns\ where table_name='products'") column_names = [row[0] for row in cursor] print("Column names:\n") for i in column_names: print(i)
Producción:
Column names: product_no name price
Publicación traducida automáticamente
Artículo escrito por isitapol2002 y traducido por Barcelona Geeks. The original can be accessed here. Licence: CCBY-SA