如何在C++中实现类似Python np.concatenate(axis=1)的二维向量列向拼接？

嘿，我完全懂你的需求——你想实现列向拼接二维向量，就像Python里np.concatenate([x,y,z], axis=1)那样，但现在用insert得到的是行向拼接，结果和预期不符。咱们来解决这个问题，有两种常用的思路，我都给你详细讲讲~

为什么你的现有代码是行向拼接？ #

先搞清楚问题出在哪：dest.insert(dest.end(), src.begin(), src.end())这段代码，是把src里的每一行（也就是每个std::vector<int>）直接追加到dest的行列表末尾，相当于把两个矩阵按行堆叠，自然得到的是行向拼接的结果。

方法一：转置+行拼接+再转置 #

你提到了转置的思路，这确实是个通用的办法，逻辑和NumPy里的思路一致：把列拼接转换成行拼接来做，步骤如下：

1. 把所有要拼接的二维向量转置（行变列，列变行）；
2. 对转置后的向量做你原来的行拼接操作；
3. 把拼接后的结果再次转置，就得到了列向拼接的最终结果。

先写一个通用的转置函数：

#include <vector>
#include <iostream>

// 转置二维整数向量的通用函数
std::vector<std::vector<int>> transpose(const std::vector<std::vector<int>>& matrix) {
    if (matrix.empty() || matrix[0].empty()) {
        return {}; // 空矩阵直接返回空
    }
    size_t original_rows = matrix.size();
    size_t original_cols = matrix[0].size();
    
    // 创建转置后的矩阵：原列数变行数，原行数变列数
    std::vector<std::vector<int>> transposed(original_cols, std::vector<int>(original_rows));
    
    for (size_t i = 0; i < original_rows; ++i) {
        for (size_t j = 0; j < original_cols; ++j) {
            transposed[j][i] = matrix[i][j];
        }
    }
    return transposed;
}

然后用这个函数实现你的需求：

int main() {
    std::vector<std::vector<int>> dest{{1,2,3,4,5}};
    std::vector<std::vector<int>> src{{6,7,8,9,10}};
    
    // 1. 转置两个矩阵
    auto dest_transposed = transpose(dest);
    auto src_transposed = transpose(src);
    
    // 2. 行拼接转置后的矩阵
    dest_transposed.insert(dest_transposed.end(), src_transposed.begin(), src_transposed.end());
    
    // 3. 再次转置得到列拼接结果
    auto final_result = transpose(dest_transposed);
    
    // 输出验证
    for (const auto& row : final_result) {
        for (int num : row) {
            std::cout << num << " ";
        }
        std::cout << "\n";
    }
    // 输出结果：1 6 2 7 3 8 4 9 5 10
    return 0;
}

这个方法的好处是：如果要拼接多个矩阵（比如x、y、z三个），只需要把所有矩阵转置后一次性行拼接，再转置回来就行，逻辑清晰，适合矩阵数量多的场景。

方法二：直接遍历行进行列拼接 #

如果不想做两次转置，也可以直接操作每一行，把源矩阵的对应行元素追加到目标矩阵的对应行末尾——这种方法更高效，因为少了两次转置的开销，前提是所有要拼接的矩阵行数必须相同（就像NumPy里axis=1要求输入矩阵的行数一致）。

写一个通用的列拼接函数：

#include <vector>
#include <iostream>
#include <cerrno>

// 列向拼接函数：把src的列追加到dest的列后面
void concatenate_columns(std::vector<std::vector<int>>& dest, const std::vector<std::vector<int>>& src) {
    // 先检查行数是否匹配，不匹配直接报错
    if (dest.size() != src.size()) {
        std::cerr << "Error: 无法列拼接！两个矩阵的行数必须相同。\n";
        return;
    }
    
    // 遍历每一行，把src的当前行元素追加到dest当前行的末尾
    for (size_t i = 0; i < dest.size(); ++i) {
        dest[i].insert(dest[i].end(), src[i].begin(), src[i].end());
    }
}

然后用这个函数实现你的需求：

int main() {
    std::vector<std::vector<int>> dest{{1,2,3,4,5}};
    std::vector<std::vector<int>> src{{6,7,8,9,10}};
    
    concatenate_columns(dest, src);
    
    // 输出验证
    for (const auto& row : dest) {
        for (int num : row) {
            std::cout << num << " ";
        }
        std::cout << "\n";
    }
    // 输出结果：1 6 2 7 3 8 4 9 5 10
    return 0;
}

这个方法更直接，性能也更好，适合大多数日常场景，推荐优先使用~

注意事项 #

• 不管用哪种方法，都要确保所有参与拼接的矩阵行数一致（如果是列拼接的话），否则会出现维度不匹配的问题，就像NumPy里会抛出错误一样；
• 如果矩阵里的元素类型不是int，只需要把函数里的int换成对应的类型（比如double）即可，通用性很强。

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