如何为数据表格添加包含指定行乘积的新行？

问题描述 #

原始数据：

ID X1 X2 X3 X4
A  3  4  5  6 
B  5  7  9  2
C  8  5  2  6

需要追加两行，分别计算A与B、B与C对应列的乘积，得到目标数据：

ID X1 X2 X3 X4
A  3  4  5  6 
B  5  7  9  2
C  8  5  2  6
AB 15 28 45 12
BC 40 35 18 12

方法一：使用Python Pandas实现 #

适合批量数据处理，灵活性高：

import pandas as pd

# 构造或读取原始数据（文件读取可用pd.read_csv）
data = {
    'ID': ['A', 'B', 'C'],
    'X1': [3, 5, 8],
    'X2': [4, 7, 5],
    'X3': [5, 9, 2],
    'X4': [6, 2, 6]
}
df = pd.DataFrame(data)

# 计算AB行：A与B对应列的乘积
ab_row = pd.DataFrame({
    'ID': ['AB'],
    'X1': [df.loc[df['ID'] == 'A', 'X1'].iloc[0] * df.loc[df['ID'] == 'B', 'X1'].iloc[0]],
    'X2': [df.loc[df['ID'] == 'A', 'X2'].iloc[0] * df.loc[df['ID'] == 'B', 'X2'].iloc[0]],
    'X3': [df.loc[df['ID'] == 'A', 'X3'].iloc[0] * df.loc[df['ID'] == 'B', 'X3'].iloc[0]],
    'X4': [df.loc[df['ID'] == 'A', 'X4'].iloc[0] * df.loc[df['ID'] == 'B', 'X4'].iloc[0]]
})

# 计算BC行：B与C对应列的乘积
bc_row = pd.DataFrame({
    'ID': ['BC'],
    'X1': [df.loc[df['ID'] == 'B', 'X1'].iloc[0] * df.loc[df['ID'] == 'C', 'X1'].iloc[0]],
    'X2': [df.loc[df['ID'] == 'B', 'X2'].iloc[0] * df.loc[df['ID'] == 'C', 'X2'].iloc[0]],
    'X3': [df.loc[df['ID'] == 'B', 'X3'].iloc[0] * df.loc[df['ID'] == 'C', 'X3'].iloc[0]],
    'X4': [df.loc[df['ID'] == 'B', 'X4'].iloc[0] * df.loc[df['ID'] == 'C', 'X4'].iloc[0]]
})

# 合并数据并输出
result_df = pd.concat([df, ab_row, bc_row], ignore_index=True)
print(result_df.to_string(index=False))
# 保存到文件可用：result_df.to_csv('output.txt', index=False, sep='  ')

方法二：使用Excel实现 #

手动公式法 #

假设原始数据位于A1:E4区域：

1. 在A5单元格输入AB，然后：
   • B5单元格输入公式：=VLOOKUP("A",$A$1:$E$4,2,FALSE)*VLOOKUP("B",$A$1:$E$4,2,FALSE)
   • C5单元格输入公式：=VLOOKUP("A",$A$1:$E$4,3,FALSE)*VLOOKUP("B",$A$1:$E$4,3,FALSE)
   • D5、E5单元格同理，将公式中的数字3、4依次替换为4、5
2. 在A6单元格输入BC，将上述公式中的"A"替换为"B"、"B"替换为"C"，复制到对应列即可

Power Query自动化法 #

1. 选中数据区域，点击「数据」选项卡的「从表格/范围」导入Power Query编辑器
2. 复制原始表，分别筛选出A+B行、B+C行，对数值列执行相乘计算，重命名ID为AB、BC
3. 将新生成的两行追加回原始表，最后加载回Excel

方法三：使用Bash脚本（Linux/macOS终端） #

假设原始数据保存为input.txt，执行以下脚本：

awk '
BEGIN {OFS="  "}
NR==1 {header=$0; print; next}
NR==2 {a1=$2; a2=$3; a3=$4; a4=$5; print}
NR==3 {b1=$2; b2=$3; b3=$4; b4=$5; print}
NR==4 {c1=$2; c2=$3; c3=$4; c4=$5; print; 
        print "AB", a1*b1, a2*b2, a3*b3, a4*b4; 
        print "BC", b1*c1, b2*c2, b3*c3, b4*c4}
' input.txt

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