c++使用vector实现按列求和
#include <vector>
#include <iostream>
using namespace std;
vector<vector<double>> colwise_sum(vector<vector<double>> m) {
vector<vector<double>> array;
vector<double> tmp;
for (int i = 0; i < m[0].size(); i++)
{
double sum = 0;
for (int j = 0; j < m.size(); j++)
{
sum += m[j][i];
}
tmp.push_back(sum);
}
array.push_back(tmp);
return array;
}
int main() {
vector<vector<double>> nums1 = {
{1, 2, 3},
{4, 15, 6},
{7, 8, 8}
};
vector<vector<double>> w = colwise_sum(nums1);
for (int i = 0; i < w.size(); i++)
{
for (int j = 0; j < w[0].size(); j++)
{
cout << w[i][j] << " ";
}
cout << endl;
}
}