编写一个基本的学校作业示例，从Swift到Python，我在简化方面遇到了麻烦

// 定义学生结构体
struct Student {
    var name: String
    var subject: String
    var score: Double
}

// 创建学生数组
var students = [Student]()

// 添加学生函数
func addStudent(name: String, subject: String, score: Double) {
    let student = Student(name: name, subject: subject, score: score)
    students.append(student)
}

// 查询学生信息函数
func getStudentInfo(name: String) -> Student? {
    for student in students {
        if student.name == name {
            return student
        }
    }
    return nil
}

// 计算平均成绩函数
func calculateAverageScore() -> Double {
    var totalScore = 0.0
    for student in students {
        totalScore += student.score
    }
    return totalScore / Double(students.count)
}

// 添加学生
addStudent(name: "张三", subject: "数学", score: 90.0)
addStudent(name: "李四", subject: "英语", score: 85.0)

// 查询学生信息
if let student = getStudentInfo(name: "张三") {
    print("姓名：\(student.name)，科目：\(student.subject)，成绩：\(student.score)")
} else {
    print("未找到该学生信息")
}

// 计算平均成绩
let averageScore = calculateAverageScore()
print("平均成绩：\(averageScore)")

# 定义学生类
class Student:
    def __init__(self, name, subject, score):
        self.name = name
        self.subject = subject
        self.score = score

# 创建学生列表
students = []

# 添加学生函数
def add_student(name, subject, score):
    student = Student(name, subject, score)
    students.append(student)

# 查询学生信息函数
def get_student_info(name):
    for student in students:
        if student.name == name:
            return student
    return None

# 计算平均成绩函数
def calculate_average_score():
    total_score = 0.0
    for student in students:
        total_score += student.score
    return total_score / len(students)

# 添加学生
add_student("张三", "数学", 90.0)
add_student("李四", "英语", 85.0)

# 查询学生信息
student = get_student_info("张三")
if student:
    print(f"姓名：{student.name}，科目：{student.subject}，成绩：{student.score}")
else:
    print("未找到该学生信息")

# 计算平均成绩
average_score = calculate_average_score()
print(f"平均成绩：{average_score}")