import numpy as np

学生信息数组

data = np.array([['大一', '李宏卓', 18, '男', 175, 65], ['大二', '李思真', 19, '女', 165, 60], ['大三', '张振海', 20, '男', 178, 70], ['大四', '赵鸿飞', 21, '男', 175, 75], ['大二', '白蓉', 19, '女', 160, 55], ['大三', '马腾飞', 20, '男', 180, 70], ['大一', '张晓凡', 18, '女', 167, 52], ['大三', '金紫萱', 20, '女', 170, 53], ['大四', '金烨', 21, '男', 185, 73]])

以年级信息为分组键，对学生信息进行分组

groups = {} for row in data: grade = row[0] if grade not in groups: groups[grade] = [] groups[grade].append(row)

输出大一学生信息

print("大一学生信息：") for row in groups['大一']: print(row)

分别计算四个年级中身高最高的同学

print("四个年级中身高最高的同学：") for grade in ['大一', '大二', '大三', '大四']: group_data = groups[grade] max_height = 0 max_student = None for row in group_data: if row[4] > max_height: max_height = row[4] max_student = row print("{}年级身高最高的同学是：{}".format(grade, max_student[1]))

计算大一学生与大三学生的平均体重

grade1_data = groups['大一'] grade3_data = groups['大三'] grade1_weights = [row[5] for row in grade1_data] grade3_weights = [row[5] for row in grade3_data] grade1_avg_weight = np.mean(grade1_weights) grade3_avg_weight = np.mean(grade3_weights) print("大一学生的平均体重为：{:.2f}kg".format(grade1_avg_weight)) print("大三学生的平均体重为：{:.2f}kg".format(grade3_avg_weight)