python 连接mysql查询数据库主键
时间: 2023-08-16 09:15:59 浏览: 342
要使用Python连接MySQL数据库并查询主键,你可以使用Python的MySQL连接库(例如`mysql-connector-python`或`pymysql`)来实现。以下是一个示例代码:
```python
import mysql.connector
# 建立数据库连接
cnx = mysql.connector.connect(
host="localhost",
user="your_username",
password="your_password",
database="your_database"
)
# 创建游标对象
cursor = cnx.cursor()
# 执行查询语句
query = "SELECT COLUMN_NAME FROM INFORMATION_SCHEMA.KEY_COLUMN_USAGE WHERE TABLE_NAME = 'your_table_name' AND CONSTRAINT_NAME = 'PRIMARY'"
cursor.execute(query)
# 获取查询结果
result = cursor.fetchall()
# 打印主键列名
for column_name in result:
print(column_name[0])
# 关闭游标和连接
cursor.close()
cnx.close()
```
请替换代码中的`your_username`、`your_password`、`your_database`和`your_table_name`为相应的数据库信息。这段代码使用了`INFORMATION_SCHEMA.KEY_COLUMN_USAGE`系统视图来查询主键列名。通过循环打印`result`中的结果,您可以获得主键的列名。
请注意,您需要安装合适的MySQL连接库,可以使用以下命令安装:
```shell
pip install mysql-connector-python
```
或
```shell
pip install pymysql
```
希望对你有所帮助!如果你还有其他问题,请随时提问。
阅读全文