拆分

水平拆分矩阵，对每个新数组应用函数，并将结果向量堆叠在一起：

npairs = data.shape[1] // 2
func = lambda x: (x*x).sum(1, keepdims=True)
answer = np.hstack([func(arr) for arr in np.hsplit(data, npairs)])

字符串

整形

在整形时，我们使用两个新索引在水平轴上迭代，其中最后一个索引是2，以便将列分成对。然后我们沿着最后一个轴应用所需的函数：

arr = data.reshape(data.shape[0], -1, 2)
answer = (arr*arr).sum(-1)

型
备注：

• data.reshape(x, -1, y)表示第一个和最后一个尺寸设置为x和y，中间值自动计算
• data.sum(-1)表示沿着最后一个轴计算总和