如何在函数中使用for循环返回元素？

def get_elements():
    elements = []
    for i in range(1, 5):
        elements.append(i)
    return elements

result = get_elements()
print(result)  # 输出：[1, 2, 3, 4]

def get_elements():
    for i in range(1, 5):
        yield i

result = get_elements()
for element in result:
    print(element)  # 输出：1, 2, 3, 4

#include <iostream>
#include <array>

std::array<int, 4> get_elements() {
    std::array<int, 4> elements;
    for (int i = 0; i < 4; i++) {
        elements[i] = i + 1;
    }
    return elements;
}

int main() {
    std::array<int, 4> result = get_elements();
    for (int element : result) {
        std::cout << element << " ";  // 输出：1 2 3 4
    }
    return 0;
}