如何用1到8的循环替换代码中DF名称和过滤条件里的数字7？

直接创建df_sample_1、df_sample_2这类零散变量会显得杂乱，建议用字典来存储不同quality对应的结果，既整洁又方便后续调用。以下是实现代码：

# 初始化字典存储各quality的样本DataFrame
df_samples = {}
# 初始化字典存储各quality的均值统计结果
df_sample_means = {}

# 循环遍历1到8的quality值
for q in range(1, 9):
    # 过滤对应quality的样本
    df_samples[q] = df[df["quality"] == q]
    # 计算统计均值并删除不需要的列
    mean_df = df_samples[q].describe().T
    mean_df = mean_df.drop(columns=["count","std","min","25%","75%","max"], axis=1)
    df_sample_means[q] = mean_df

使用说明 #

• 要获取quality=7的样本DataFrame：df_samples[7]
• 要获取quality=7的均值统计结果：df_sample_means[7]

内容的提问来源于stack exchange，提问作者arturo oliva