如何对数组中相同索引的元素进行聚合运算

需求说明 #

需要将带时间戳的日期字段提取纯日期作为分组依据，把同一日期下对应数组的每个元素逐个累加，生成合并后的数组，同时适配多元素（如16个及更多）的数组批量处理。

实现步骤（以Python为例） #

1. 数据解析 #

从原始表格提取DateTime和Column A数据，拆分DateTime得到纯日期，将数组字符串转换为整数列表，方便后续计算。

2. 分组累加 #

用字典按日期分组，遍历每个分组内的数组，对对应位置的元素完成累加操作。

3. 结果格式化 #

将累加后的数组转回字符串格式，重新整理成目标表格的HTML结构。

代码实现 #

# 模拟从表格中提取的原始数据
raw_data = [
    {"DateTime": "4/1/2022 17:00", "Column A": "[10,5,1,0,2]"},
    {"DateTime": "4/1/2022 18:00", "Column A": "[5,2,0,1,3]"},
    # 测试16元素的数组场景
    {"DateTime": "4/2/2022 09:00", "Column A": "[1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16]"},
    {"DateTime": "4/2/2022 14:00", "Column A": "[16,15,14,13,12,11,10,9,8,7,6,5,4,3,2,1]"}
]

# 初始化分组字典，存储每个日期的累加数组
date_accum = {}

for entry in raw_data:
    # 提取纯日期部分
    date = entry["DateTime"].split(" ")[0]
    # 解析数组字符串为整数列表
    num_list = list(map(int, entry["Column A"].strip("[]").split(",")))
    
    if date not in date_accum:
        # 首次出现该日期，直接存入数组
        date_accum[date] = num_list.copy()
    else:
        # 对应位置元素累加，自动适配数组长度
        for idx in range(len(num_list)):
            date_accum[date][idx] += num_list[idx]

# 生成目标格式的HTML表格
output_html = """<div class="s-table-container">
<table class="s-table">
<thead>
<tr>
<th>Date</th>
<th>Column A</th>
</tr>
</thead>
<tbody>
"""
for date, final_arr in date_accum.items():
    # 将整数列表转回带[]的字符串格式
    arr_str = f"[{','.join(map(str, final_arr))}]"
    output_html += f"<tr><td>{date}</td><td>{arr_str}</td></tr>\n"

output_html += """</tbody>
</table>
</div>"""

print(output_html)

关键细节说明 #

• 日期提取：通过字符串拆分split(" ")[0]快速获取纯日期，无需复杂日期解析库（适用于日期格式统一的场景）。
• 数组解析：用strip("[]")去除数组首尾括号，split(",")拆分元素后转成整数列表，不管数组有多少元素都能通用处理。
• 累加逻辑：利用字典分组，遍历数组时逐个位置相加，只要同一日期下的数组元素数量一致，就能自动完成批量累加，无需手动处理每个元素。

运行结果 #

执行代码后，会输出符合要求的HTML表格。比如示例中4/1/2022的两条数据会合并为[15,7,1,1,5]，4/2/2022的16元素数组会合并为[17,17,17,17,17,17,17,17,17,17,17,17,17,17,17,17]。

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