Google ColabでCSVを読み込む4つの方法

2020-02-19
プログラミング

URLから直接読み込む

from urllib.request import urlopen
from io import StringIO
import csv

data = urlopen("https://archive.ics.uci.edu/ml/machine-learning-databases/wine-quality/winequality-red.csv").read().decode()
csvReader = csv.reader(StringIO(data), delimiter=";")

for row in csvReader:
  print(row)

ローカルファイルから読み込み

import csv
from google.colab import files
uploaded = files.upload()

ファイルをアップロードする

with open("winequality-red.csv") as fp:
  for row in csv.reader(fp, delimiter=";"):
    print(row)

Google Driveをマウントする

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

認証コードを登録する

%%bash
cd '/content/gdrive/My Drive'
with open("./winequality-red.csv") as fp:
  for row in csv.reader(fp, delimiter=";"):
    print(row)

Google Driveから読み取る方法

from google.colab import auth
auth.authenticate_user()

from pydrive.drive import GoogleDrive
from pydrive.auth import GoogleAuth
from oauth2client.client import GoogleCredentials
gauth = GoogleAuth()
gauth.credentials = GoogleCredentials.get_application_default()
drive = GoogleDrive(gauth)

認証コードを登録する

file_data = drive.CreateFile({'id': 'シェアリンクのID'})
file_data.GetContentFile('winequality-red.csv')
with open("winequality-red.csv") as fp:
  for row in csv.reader(fp, delimiter=";"):
    print(row)