Pythonでエクセルの操作

Pythonでエクセルの操作をしたい場合は、pandasというライブラリが使えます。
Excelの読み込み → 操作 → 書き込みの流れでよく使われる最低限の関数をまとめました。

Excelの読み込み

import pandas as pd

path = 'sample.xlsx'
df = pd.read_excel(path)

df_queried = df_target.query('name == [u"Anna",u"Erza"]')

• read_excel: エクセルを読む、DataFrame型を返す 公式Doc
• DataFrame.query: 読んだエクセルのデータを絞り込む 公式Doc

Excelの操作

for index, row in df_target_queried.iterrows():
   print row['name'] # name列の要素を取得する
   print row[u'名前'] # 日本語の場合Unicode指定が必要な場合も

for column in df_target_queried:
   print column # 列ごとに展開する

• DataFrame.iterrows(): 読んだエクセルのデータを列ごとに展開するループ関数

Excelの書き込み

out = {
    'name': ['Erza', 'Anna'],
    'age': [22, 25],
    'color': ['red', 'pink']
}

df_out = pd.DataFrame(out, columns=['name', 'age', 'color'])
writer = pd.ExcelWriter("save.xlsx")
df_out.to_excel(writer, sheet_name="save")

writer.save()

• ExcelWriter: pandaのExcel書き込みオプジェクトを作成する。 DataFrame型を渡す必要がある　公式Doc
• to_excel: Excelのワークブック上のシートにデータを書き込む 公式Doc