Pymysql将MySQL数据导出并分类
时间: 2024-02-22 17:00:56 浏览: 121
您可以使用Pymysql连接MySQL数据库并执行查询语句来导出数据。要分类数据,您需要在查询语句中包含筛选条件或使用GROUP BY子句。
例如,以下代码可以将名为“example_table”的表中的所有数据导出到名为“example.csv”的CSV文件中:
```python
import csv
import pymysql
# 连接到MySQL数据库
conn = pymysql.connect(host='localhost', port=3306, user='root', passwd='password', db='example_db')
# 执行查询语句
cursor = conn.cursor()
cursor.execute("SELECT * FROM example_table")
# 将数据写入CSV文件
with open('example.csv', 'w', newline='') as file:
writer = csv.writer(file)
writer.writerow([i[0] for i in cursor.description]) # 写入列名
writer.writerows(cursor)
# 关闭数据库连接
conn.close()
```
要按某些条件分类数据,例如按“category”列分类,您可以修改查询语句:
```python
cursor.execute("SELECT * FROM example_table WHERE category='example_category'")
```
或者使用GROUP BY子句:
```python
cursor.execute("SELECT category, COUNT(*) FROM example_table GROUP BY category")
```
这将返回一个结果集,其中每行包含一个分类和该分类中的行数。您可以使用类似上面的代码将结果写入CSV文件。
阅读全文