【4 weeks challenge – Python】Day 22 指定のkeyだけCSVに出力

daily_trial

こんにちは、しらすです。

22日目です。本日の習得スキルは「指定のkeyだけCSVに出力」です!

指定のkeyだけCSVに出力でできること

指定のkeyだけを切り出し、新しくcsvファイルを作成することができます。単純なcsv整理であればこれで十分です。

使い方

import pandas as pd

#pandasでcsvから読み込む
csv_input = pd.read_csv("./day21_house/train.csv")

#指定の列だけ取り出して出力する
result0 = csv_input['Id']
result1 = csv_input['SalePrice']
result2 = csv_input['LotFrontage']
result3 = csv_input['LotArea']
result4 = csv_input['Street']

output = 'Id,SalePrice,LotFrontage,LotArea,Street\n'
for i in range(len(result1)):
    output = output \
            + str(result0[i]) + ','\
            + str(result1[i]) + ','\
            + str(result2[i]) + ','\
            + str(result3[i]) + ','\
            + str(result4[i]) + '\n'

with open('./day22/temp.csv','w') as file:
    file.write(output)

結果

コメント