如何使用现有数组和公式借助Numpy创建新数组？

实现步骤 #

• 先导入Numpy库，这是使用所有Numpy功能的基础
• 将普通Python列表转换为Numpy数组，Numpy的核心优势是向量化运算，针对数组的操作会自动作用于每一个元素
• 直接应用公式y=2x，Numpy会自动完成元素级的乘法运算

完整代码示例 #

# 导入Numpy库
import numpy as np

# 将原始列表转为Numpy数组
X = np.array([1, 2, 3, 4])

# 应用公式生成新数组
Y = 2 * X

# 验证结果
print(Y)  # 输出: [2 4 6 8]

补充说明 #

Numpy的向量化运算不需要手动写循环遍历元素，这种写法不仅代码更简洁，处理大规模数组时的执行效率也远高于普通Python循环。如果需要更复杂的公式（比如y=3x²+2x+1），同样可以直接通过数组运算实现：

Y = 3 * X**2 + 2 * X + 1

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