[python] データ読み込み

pythonで使えるファイル読み込み方法をいくつかあげました。
どんな関数があったっけ、となった時用なので、各関数の詳しい説明は省いてます。

open()

ファイルとモードを指定。
モードのデフォルトは'r'(読み込み専用)

main.py

#一行ずつ読み込んで表示する場合
data = open('/path/to/data', 'r')

for line in data:
  print line

#下も同じ操作
#line = f.readline()

data.close()



#すべての内容を読み込んで表示する場合
data = open('/path/to/data', 'r')

#文字列にして返す場合
all_data = data.read()
print all_data

#リストにして返す場合
#all_data_list = f.readlines()
#print all_data_list

data.close()

このファイル操作は開く操作と閉じる操作が必要になる。閉じるのを忘れてしまいがちだが、以下の場合は閉じるコマンドが必要ないので安心

main.py

with open(file_path) as f:
    for line in f:
        ###command

read_csv(), read_table()

pandas
カンマ区切りならread_csv(), タブ区切りならread_table()
読み込んだものはデータフレームとして返される

main.py

import pandas as pd

data_csv = pd.read_csv('/path/to/data.csv')
data_tsv = pd.read_table('/path/to/data.tsv')

デフォルトはheaderありなので、ファイルにheaderがない場合はheader=Noneもしくはnames=()で指定する

loadtxt()

numpy
区切り文字と型も指定する(デフォルトはスペース区切りとfloat)
読み込んだものは配列(array)として返される

main.py

import numpy as np

data = np.loadtxt('/path/to/data', delimiter=' ', dtype = 'float')