如何在特定列中取每3行的平均值？

import pandas as pd

# 读取数据集，假设数据集存储在名为"dataset.csv"的文件中
data = pd.read_csv("dataset.csv")

# 确定要操作的特定列
column_name = "column_name"

# 按照特定列进行排序
sorted_data = data.sort_values(by=column_name)

# 存储每3行的平均值
averages = []

# 遍历数据集
for i in range(0, len(sorted_data), 3):
    # 取当前行及其后两行的特定列数值
    values = sorted_data[column_name].iloc[i:i+3]
    
    # 计算平均值并存储
    average = values.mean()
    averages.append(average)

# 打印结果
print(averages)