如何根据给定x数组和多项式公式生成x与y配对的二维数组

下面以Python环境为例，提供几种常用实现方式：

方法1：列表推导式（最简洁，适合常规场景） #

直接遍历x数组的每个元素，同步计算对应y值并拼接成子列表即可，代码如下：

x = [-10, -7, 2, 6, 11, 12]
paired_arr = [[val, 3 * val ** 2 + 2 * val - 1] for val in x]

运行后paired_arr的输出就是你需要的格式：

[[-10, 279], [-7, 132], [2, 15], [6, 119], [11, 384], [12, 455]]

方法2：函数式写法（基于map） #

如果你习惯函数式编程风格，可以用map函数实现：

x = [-10, -7, 2, 6, 11, 12]
# 先定义y的计算函数
calc_y = lambda x_val: 3 * x_val ** 2 + 2 * x_val - 1
paired_arr = list(map(lambda val: [val, calc_y(val)], x))

方法3：Numpy实现（适合大数据量批量计算） #

如果处理的x数组数据量很大，用numpy的向量化运算效率更高：

import numpy as np
x = np.array([-10, -7, 2, 6, 11, 12])
y = 3 * x ** 2 + 2 * x - 1
# 按列拼接后转成普通列表
paired_arr = np.column_stack((x, y)).tolist()

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