python - パンダは集計後に保存する列を選択します

Question

データを読み取り、グループの平均を計算してから、データを他のファイルに保存します。問題は、そのファイルに書き込む列の一部のみを選択したいということです。現在、エラーが発生するため、すべてをファイルに保存してから、再度開き、列を選択して再度書き込む必要があります。

このコードは機能しません:

df = pd.read_csv("./file.csv")
grouped = df.groupby(['Condition','ID','BeginEnd'])
agg = grouped.mean()
agg = agg[['Condition','ID','BeginEnd3','KnowledgeScore']]
agg.to_csv('file2.csv')

私はこれをしなければなりません：

df = pd.read_csv("./file.csv")
grouped = df.groupby(['Condition','ID','BeginEnd'])
agg = grouped.mean()
agg.to_csv('file2.csv')
df3 = pd.read_csv('file2.csv')
df3 = df3[['Condition','ID','BeginEnd3','KnowledgeScore']]
df3.to_csv('file2.csv')