如何在一行代码中重命名Pandas列并将代码压缩至4行完成多目标

需求概述 #

需用4行以内代码完成以下操作：

• 创建含Name、Year 1、Year 2三列的空DataFrame
• 将列名重命名为Project、y23、y24
• 将DataFrame保存为data/df.csv
• 读取CSV时指定索引列
• 打印读取后的结果

原始代码 #

df = pd.DataFrame(columns = ['Name', 'Year 1','Year 2'])
df = df.rename(columns = {'Name':'Project', 'Year 1)':'y23','Year 2':'y24'})
df.to_csv('data/df.csv')
df2 = pd.read_csv('data/df.csv')
print(df2)

优化修正后的代码 #

原提交的代码存在两处问题：一是重命名字典里的'Year 1)'多了多余的右括号，二是读取CSV时未按要求指定索引列。以下是符合要求的4行完整代码：

# 第1行：创建空DataFrame
df = pd.DataFrame(columns=['Name', 'Year 1', 'Year 2'])
# 第2行：重命名列并保存为CSV（指定index=False避免生成无意义索引列）
df.rename(columns={'Name':'Project', 'Year 1':'y23', 'Year 2':'y24'}).to_csv('data/df.csv', index=False)
# 第3行：读取CSV，因保存时未存索引，直接读取即可；若保存时带索引，需用index_col=0指定
df2 = pd.read_csv('data/df.csv')
# 第4行：打印结果
print(df2)

如果要把创建、重命名、保存合并为一行，也可以写成3行代码（满足4行以内要求）：

pd.DataFrame(columns=['Name', 'Year 1', 'Year 2']).rename(columns={'Name':'Project', 'Year 1':'y23', 'Year 2':'y24'}).to_csv('data/df.csv', index=False)
df2 = pd.read_csv('data/df.csv')
print(df2)

内容的提问来源于stack exchange，提问作者Mr. Student.exe