使用Python采集数据库通常涉及以下步骤:
选择合适的库
对于MySQL数据库,可以使用`pymysql`或`SQLAlchemy`。
对于SQLite数据库,可以使用Python的标准库`sqlite3`。
建立数据库连接
使用库提供的函数建立与数据库的连接。例如,使用`pymysql.connect`连接到MySQL数据库。
执行SQL查询
通过游标对象执行SQL查询语句。
处理查询结果
使用游标的`fetchall()`或`fetchone()`方法获取查询结果。
关闭数据库连接
完成数据提取后,关闭数据库连接以释放资源。
import pymysql
连接数据库
def connect_db(host, user, password, db_name, charset='utf8'):
try:
db = pymysql.connect(host=host, user=user, password=password, db=db_name, charset=charset)
return db
except Exception as e:
print(f"数据库连接错误: {e}")
return None
执行SQL查询并获取结果
def get_data_from_db(db, query):
try:
with db.cursor() as cursor:
cursor.execute(query)
result = cursor.fetchall()
return result
except Exception as e:
print(f"查询错误: {e}")
return None
关闭数据库连接
def close_db(db):
if db:
db.close()
主程序
if __name__ == "__main__":
数据库连接信息
host = '192.168.1.110'
user = 'unickcheng'
password = 'p123'
db_name = 'UNICK'
建立连接
db = connect_db(host, user, password, db_name)
SQL查询语句
sql = "SELECT * FROM your_table_name"
获取数据
data = get_data_from_db(db, sql)
关闭连接
close_db(db)
打印数据
for row in data:
print(row)
请根据你的具体需求修改数据库连接信息和SQL查询语句。如果你需要处理更复杂的数据处理任务,比如数据清洗、转换或存储,可以使用`pandas`库来辅助操作。