どのようにエクセルのデータを辞書に入れるか？

Question

エクセルのデータを辞書に入れたいです。
エクセルは、

のようになっています。
views.py に

#coding:utf-8
from django.shortcuts import render
import xlrd

book3 = xlrd.open_workbook('./data/excel.xlsx')
sheet3 = book3.sheet_by_index(0)

large_item = None
data_dict = {}
for row_index in range(1,sheet3.nrows):
    rows3 = sheet3.row_values(row_index)
    large_item = rows3[1] or large_item
    data_dict = rows3

のように書きました。

print(data_dict)

の結果は['', '4', '10', 'Karen', ''] のようになりました。
この前は、

data_dict = rows3

を

data_dict.extend(rows3)

のように書いていたのですが、dict has not extend というエラーが発生しました。

data_dict = {
    1: {
        user_id: 1,
        name_id: 1,
        name: Blear,
        age: 40,
        man: false,
        employee: leader,
    },
    2: {
        user_id: 2,
        name_id: 5,
           ・
       ・
       ・
    },
           ・
       ・
       ・
}

のようなアウトプットにしたいのですが、どうしたら良いでしょうか？

Answer 1

こんな感じでしょうか。

import xlrd
from pprint import pprint

def try_to_int(arg):
    try:
        return int(arg)
    except:
        return arg

def main():
    book3 = xlrd.open_workbook('./data/excel.xlsx')
    sheet3 = book3.sheet_by_index(0)

    data_dict = {}
    tag_list = sheet3.row_values(0)[1:]
    for row_index in range(1, sheet3.nrows):
        row = sheet3.row_values(row_index)[1:]
        row = list(map(try_to_int, row))
        data_dict[row_index] = dict(zip(tag_list, row))

    pprint(data_dict)

if __name__ == '__main__':
    main()

"""出力
{1: {'age': 40,
     'empleyee': 'leader',
     'man': '',
     'name': 'Blear',
     'name_id': 1,
     'user_id': 1},
 2: {'age': 23,
     'empleyee': 'others',
     'man': '●',
     'name': 'Tom',
     'name_id': 5,
     'user_id': 2},
 3: {'age': 52,
     'empleyee': 'manager',
     'man': '',
     'name': 'Rose',
     'name_id': 9,
     'user_id': 3},
 4: {'age': '',
     'empleyee': '',
     'man': '',
     'name': 'Karen',
     'name_id': 10,
     'user_id': 4}}
"""

large_itemは仕様が分からなかったのでスルーしてます。
xlrdは初めて使ったので、慣習などは知りません。ご容赦ください。

---

いい練習になると思って、ジェネレータを使ってみた。

import xlrd
from pprint import pprint

def try_to_int(arg):
    try:
        return int(arg)
    except:
        return arg

def read_all_row(worksheet, left_up=(0, 0)):
    for i in range(left_up[0], worksheet.nrows):
        ret_list = worksheet.row_values(i)[left_up[1]:]
        yield list(map(try_to_int, ret_list))

def main():
    book3 = xlrd.open_workbook('./data/excel.xlsx')
    sheet3 = book3.sheet_by_index(0)

    data_dict = {}
    tag_list = sheet3.row_values(0)[1:]
    for i, row in enumerate(read_all_row(sheet3, left_up=(1, 1))):
        data_dict[i] = dict(zip(tag_list, row))

    pprint(data_dict)

if __name__ == '__main__':
    main()

なんかすっきりしない。ラッパークラスを作った方がいいのかもしれない。