Flask 如何在SQLAlchemy查询中使用avg和sum函数

在本文中，我们将介绍如何在Flask应用程序中使用SQLAlchemy的查询语言来使用avg和sum函数计算平均值和总和。

阅读更多：Flask 教程

创建Flask应用程序

首先，我们需要创建一个Flask应用程序。我们可以使用Flask框架快速搭建一个简单的应用程序。

from flask import Flask
from flask_sqlalchemy import SQLAlchemy

app = Flask(__name__)
app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///test.db'
db = SQLAlchemy(app)

在上面的代码中，我们创建了一个Flask应用程序，并配置了一个SQLite数据库。然后，我们引入了SQLAlchemy扩展并将其与应用程序关联。

接下来，我们将定义一个模型来表示我们的数据表。假设我们有一个名为”students”的表，包含以下字段：id（学生ID）、name（学生姓名）和score（学生成绩）。

class Student(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    name = db.Column(db.String(50))
    score = db.Column(db.Integer)

添加测试数据

在进行查询之前，我们需要向数据库中添加一些测试数据。为了简单起见，我们添加三个学生的信息。

s1 = Student(name="Alice", score=80)
s2 = Student(name="Bob", score=75)
s3 = Student(name="Cindy", score=90)

db.session.add_all([s1, s2, s3])
db.session.commit()

使用avg函数计算平均值

要在SQLAlchemy查询中使用avg函数计算平均值，我们可以使用func.avg()函数。

from sqlalchemy import func

avg_score = db.session.query(func.avg(Student.score)).scalar()

在上面的代码中，我们通过func.avg(Student.score)来计算学生成绩的平均值，并使用scalar()方法获取计算结果。

使用sum函数计算总和

要在SQLAlchemy查询中使用sum函数计算总和，我们可以使用func.sum()函数。

total_score = db.session.query(func.sum(Student.score)).scalar()

在上面的代码中，我们通过func.sum(Student.score)来计算学生成绩的总和，并使用scalar()方法获取计算结果。

示例说明

下面我们使用一个例子来说明如何使用avg和sum函数进行查询。

# 查询所有学生的成绩平均值和总和
result = db.session.query(func.avg(Student.score), func.sum(Student.score)).first()
avg = result[0]
total = result[1]

print("平均值:", avg)
print("总和:", total)

在上面的代码中，我们通过db.session.query()来创建一个查询对象，并使用func.avg()和func.sum()来计算平均值和总和。通过first()方法获取第一条记录，并使用索引来访问计算结果。