如何创建新列并通过公式赋值，同时显示size列与预测值y

别用循环折腾了，直接用Pandas的向量化运算更高效，两步就能搞定需求：

• 一键生成存储预测值的新列
• 快速提取原列与新列展示

代码示例 #

import pandas as pd

# 假设你的DataFrame是df，这里用示例数据演示
df = pd.DataFrame({'size': [3, 7, 12, 18]})

# 按公式计算预测值并创建新列
df['predicted_y'] = 0.7118 * df['size'] + 1.1691

# 展示size列和预测值列
print(df[['size', 'predicted_y']])

说明 #

Pandas的列运算默认是向量化处理，不用逐行循环遍历，数据量越大，效率优势越明显。执行代码后，df会新增predicted_y列，最后一行代码直接筛选出你需要的两列输出。

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