如何在for循环中捕获函数的输出

def my_function(x):
    return x * x

results = []
for i in range(5):
    result = my_function(i)
    results.append(result)

print(results)  # 输出: [0, 1, 4, 9, 16]

function myFunction(x) {
    return x * x;
}

let results = [];
for (let i = 0; i < 5; i++) {
    let result = myFunction(i);
    results.push(result);
}

console.log(results);  // 输出: [0, 1, 4, 9, 16]

import java.util.ArrayList;
import java.util.List;

public class Main {
    public static int myFunction(int x) {
        return x * x;
    }

    public static void main(String[] args) {
        List<Integer> results = new ArrayList<>();
        for (int i = 0; i < 5; i++) {
            int result = myFunction(i);
            results.add(result);
        }

        System.out.println(results);  // 输出: [0, 1, 4, 9, 16]
    }
}