Google Colaboratoryのファイル入出力方法

よくちょっとしたファイルをColaboratoryで集計してドライブに入出力したいときがあるのでメモ。Pythonを使用。

ドライブのマウント

実行後、ドライブにアクセス許可をもとめるダイアログがでるので許可する。

from google.colab import drive
drive.mount('/content/drive')

CSVファイルの読み込み

あらかじめドライブにおいておいたCSVファイルをpandasで読み込む。
パス、ファイル名は適宜変更する。

import pandas as pd

base_path = r"/content/drive/test_folder/"
input_path = "{}input/input_file.csv".format(base_path)

input_df = pd.read_csv(input_path)
input_df.head()

出力

集計などを行ったあと、最後にCSVで出力する。
同名のファイルを出力すると上書きされるため、出力ファイル名に日付を入れておくと便利。

from datetime import datetime, timedelta, timezone
JST = timezone(timedelta(hours=+9), 'JST')
today = datetime.now(JST).strftime("%Y%m%d%H%M")

output_file_name = "output_file_" + today + ".csv"
output_path = "{}output/{}".format(base_path, output_file_name)

output_df.to_csv(output_path)