如何在Julia中实现数组的按行求和与按列求和？

1. 如何对数组进行按行求和？ #

在Julia中，使用sum()函数并指定维度参数dims=2就能实现二维数组的按行求和。默认返回保留维度的矩阵，若需要一维数组，可通过dropdims()去除多余维度。

示例代码：

# 定义示例2×2数组
arr = [1 2; 3 4]

# 按行求和，返回2×1矩阵
row_sum = sum(arr, dims=2)
# 输出结果：
# 2×1 Matrix{Int64}:
#  3
#  7

# 转换为一维数组
row_sum_1d = dropdims(row_sum, dims=2)
# 输出结果：[3, 7]

2. 如何计算数组的按列求和（附预期结果说明） #

针对你给出的2×2数组[a b; c d]，标准按列求和的结果应为列向量[a+c; b+d]，只需在sum()中指定dims=1，再转置即可得到列向量形式：

# 定义数组（假设a、b、c、d为具体数值）
arr = [a b; c d]

# 按列求和，返回1×2矩阵
col_sum = sum(arr, dims=1)
# 转置为列向量
col_sum_col = col_sum'
# 输出结果：
# 2×1 Matrix{...}:
#  a + c
#  b + d

若你预期的结果是[a+b; c+d]，这实际是按行求和的结果，直接使用第一部分的sum(arr, dims=2)就能得到该列向量。

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