如何用Python生成包含1-100数字及其平方值的numpy数组

你给出的示例使用了Numpy库，以下是两种常用实现方案： #

方案1：Numpy原生方法（效率更高，适合大数据量场景） #

需提前安装Numpy库，实现代码如下：

import numpy as np

# 生成1到100的整数序列，注意arange为左闭右开规则，所以结束值填101
num_arr = np.arange(1, 101)
# 批量计算所有数值的平方值
square_arr = num_arr ** 2
# 按列拼接得到n行2列的目标数组
res = np.column_stack((num_arr, square_arr))

调用print(res[:5])打印前5行验证，输出结果和你给出的示例完全一致。 #

方案2：列表推导式实现（写法简单易懂，适合新手） #

如果数据量不大，也可以用列表推导式生成目标结构后转Numpy数组，代码更简洁：

import numpy as np

res = np.array([[i, i*i] for i in range(1, 101)])

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