在 C++ 中,有没有一种方法可以在向量的每个元素上调用函数,而不使用在所有向量元素上运行的循环?类似于 Python 中的“地图”。
8 回答
是的:std::for_each
。
#include <algorithm> //std::for_each
void foo(int a) {
std::cout << a << "\n";
}
std::vector<int> v;
...
std::for_each(v.begin(), v.end(), &foo);
您已经得到了几个提到std::for_each
.
虽然这些回答了您提出的问题,但我要补充一点,至少根据我的经验,std::for_each
这是标准算法中最没用的。
我使用(例如)std::transform
,它基本上a[i] = f(b[i]);
或result[i] = f(a[i], b[i]);
比std::for_each
. 许多人经常使用std::for_each
打印集合的元素;为此,std::copy
使用 astd::ostream_iterator
作为目的地会更好。
在 C++ 11 上:您可以使用 lambda。例如:
std::vector<int> nums{3, 4, 2, 9, 15, 267};
std::for_each(nums.begin(), nums.end(), [](int &n){ n++; });
如果你有 C++11,还有一个更短的方法:基于范围的 for . 它的目的正是如此。
std::vector<int> v {1,2,3,4,5};
for (int element : v)
std::cout << element; //prints 12345
您还可以在适当的时候对其应用引用和 const,或者在类型较长时使用 auto。
std::vector<std::vector<int>> v {{1,2,3},{4,5,6}};
for (const auto &vec : v)
{
for (int element : vec)
cout << element;
cout << '\n';
}
输出:
123
456
使用for_each
:
// for_each example
#include <iostream>
#include <algorithm>
#include <vector>
using namespace std;
void myfunction (int i) {
cout << " " << i;
}
struct myclass {
void operator() (int i) {cout << " " << i;}
} myobject;
int main () {
vector<int> myvector;
myvector.push_back(10);
myvector.push_back(20);
myvector.push_back(30);
cout << "myvector contains:";
for_each (myvector.begin(), myvector.end(), myfunction);
// or:
cout << "\nmyvector contains:";
for_each (myvector.begin(), myvector.end(), myobject);
cout << endl;
return 0;
}
OP 提到了map
Python 中的函数。
这个 Python 函数实际上将一个函数应用于列表(或可迭代)的每个元素,并返回一个收集所有结果的列表(或可迭代)。
换句话说,它做了这样的事情:
def f( x ) :
""" a function that computes something with x"""
# code here
return y
input = [ x1, x2, x3, ... ]
output = map( func, input )
# output is now [ f(x1), f(x2), f(x3), ...]
因此,与 Python 的映射最接近的 C++ 标准库实际上是std::transform
(来自<algorithm>
标题)。
示例用法如下:
#include <vector>
#include <algorithm>
using namespace std;
double f( int x ) {
// a function that computes the square of x divided by 2.0
return x * x / 2.0 ;
}
int main( ) {
vector<int> input{ 1, 5, 10 , 20};
vector<double> output;
output.resize( input.size() ); // unfortunately this is necessary
std::transform( input.begin(), input.end(), output.begin(), f );
// output now contains { f(1), f(5), f(10), f(20) }
// = { 0.5, 12.5, 50.0, 200.0 }
return 0;
}
您可以使用带有一对迭代器和一个函数或仿函数的std::for_each 。
以为我会分享and的std::ranges
等价物,如果有人更喜欢它们:for_each
transform
std::vector<int> v;
std::ranges::for_each(v,[](const auto& n) {});
const auto squared = v | std::views::transform([](const auto& n) { return n*2; });
在 Godbolt 上运行:https ://godbolt.org/z/zYME6b