如何计算3列平均值并根据条件新增列填充默认值

需求逻辑 #

• 计算每行中Column A、Column B、Column C三列的平均值
• 当该行平均值小于0.001时，新增的Class列填入1；反之Class列留空

原始数据表 #

处理后的目标数据表 #

实现示例（Python pandas） #

假设表格单元格为数值类型，可通过以下代码实现逻辑：

import pandas as pd

# 构造原始数据集（替换为实际读取数据的逻辑）
df = pd.DataFrame({
    'Column A': [0.0005, 0.002, 0.0008],
    'Column B': [0.0003, 0.0015, 0.0004],
    'Column C': [0.0004, 0.001, 0.0003]
})

# 计算每行三列的平均值
df['row_avg'] = df[['Column A', 'Column B', 'Column C']].mean(axis=1)

# 根据平均值条件生成Class列
df['Class'] = df['row_avg'].apply(lambda x: 1 if x < 0.001 else '')

# 移除中间计算的row_avg列（可选操作）
df = df.drop(columns=['row_avg'])

print(df)

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