[大数据]Python与mysql数据库

Python查询Mysql使用 fetchone() 方法获取单条数据, 使用fetchall() 方法获取多条数据。

• fetchone(): 该方法获取下一个查询结果集。结果集是一个对象
• fetchall(): 接收全部的返回结果行.
• rowcount: 这是一个只读属性，并返回执行execute()方法后影响的行数。

案例如下：

"""
使用pymysql将自己的姓名、年级专业、以及学号存入表中,并
且使用自己的名字条件查询，获取所有信息，打印出来，截图方式
提交
"""
import pymysql


# 数据库连接
def sql_connect():
    db = pymysql.connect(host='121.36.86.90',
                         user='root',
                         password='Zx102020!',
                         db='train')
    return db


# 查询数据
def query_sql(sql):
    db = sql_connect()
    cursor = db.cursor()
    cursor.execute(sql)
    results = cursor.fetchall()
    for row in results:
        for i in range(len(row)):
            print(row[i], end=' ')
    db.close()


# 插入数据
def insert_sql(sql):
    db = sql_connect()
    cursor = db.cursor()
    try:
        cursor.execute(sql)
        db.commit()
    except:
        db.rollback()
    db.close()


if __name__ == '__main__':
    # 插入语句
    sql1 = 'insert into student(id,name,grade,major) values(11111,"小明","20级","软件工程")'
    insert_sql(sql1)
    # 查询语句
    sql2 = 'select * from student where name = "小明"'
    query_sql(sql2)