如何将Awkward Array中各子数组的首个元素相加？

• 基础实现方式：索引提取+求和
  先导入Awkward库并创建示例数组：

  import awkward as ak
  
  my_array = ak.array([[1,2,3], [4,5,6], [7,8,9]])
  

  用[:, 0]索引提取每个子数组的第一个元素，再通过ak.sum()完成求和：

  # 提取所有子数组的首元素
  first_elements = my_array[:, 0]
  # 计算总和
  total = ak.sum(first_elements)
  print(total)  # 输出：12
  

  也可以直接合并为一行代码：

  total = ak.sum(my_array[:, 0])
  

• 处理含空数组的场景：用ak.firsts()安全提取
  如果数组中存在空的子数组，直接索引会触发错误，此时使用ak.firsts()可以安全提取首元素（空数组对应返回None，求和时会自动忽略这类值）：

  my_array_with_empty = ak.array([[1,2,3], [], [7,8,9]])
  first_elements_safe = ak.firsts(my_array_with_empty)
  total_safe = ak.sum(first_elements_safe)
  print(total_safe)  # 输出：8
  

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