Go言語でExcelファイルから特定データを抽出するアプリ作ったよ

2017/06/23

Go言語でExcelファイルを処理する方法を３回に分けて紹介しました。

何かとGo言語をdisっている当ブログですが、Excelファイルを処理するのがめちゃくちゃ楽でした（笑）のでシェア。designed by Renée French.まずはライブラリをインストールExcelをGo言語で扱えるようにするためには、tealeg/xlsxというライブラリが必要になります。コマンドプロンプトから以下のように打てばインストールでします。※GOPATHがあらかじめ設定されていることが前提です。go get github.com/tealeg/xlsx Go言語の特徴として、ライブラリを簡単にインストールできることなんですよね。これがC#だとVisual Studio上じ...

非IT企業に勤める中年サラリーマンのIT日記

Go言語でExcel処理パートⅡ シート内の値をすべて表示する

前回の記事です。前回はセルのアドレス（行No、列No）を指定して値をGETしましたが、今回はシート内の値をすべてGETしてコンソールに表示させます。designed by Renée French. 用意するExcelファイル前回同様、住所録をExcelファイルで用意しました。全部で12万行以上のデータがあります。ファイルサイズが3.5MBと大きいファイルです。ソースコードソースコードは以下の通りです。package mainimport ( "fmt" "github.com/tealeg/xlsx" //go get github.com/tealeg/xlsx)func main() { excel,...

非IT企業に勤める中年サラリーマンのIT日記

Go言語でExcel処理パートⅢ ファイルを新規作成し書き込む

Go言語でExcel処理シリーズ第三弾です。前回までの記事はこちら。前回までの２つはExcelファイルを読み込む方でしたが、今回は書き込む方についてです。ファイルを新規作成した後にセルに値を書き込むという流れとなります。ソースコード全体のソースコードは以下の通りです。package mainimport ( "fmt" "github.com/tealeg/xlsx")func main() { file := xlsx.NewFile() sheet, err := file.AddSheet("Sheet1") if err != nil { fmt.Println(err.Error()) } sheet.Cell(0, 0).Value = &quo...

これらの機能を使って、Execlファイルから特定の条件にマッチしたものだけ抽出するプログラムを作成しました。抽出したデータは別のExcelファイルに保存するようになっています。

1 各パラメーターはXMLファイルで設定
2 ソースコード
3 検索するExcelファイル
4 実行結果

各パラメーターはXMLファイルで設定

すべてのパラメーターをプログラム内で設定してしまうと汎用性がなくなるので、XMLファイルにまとめています。このパラメータを変えることで色々なExcelファイルに対応できるようになっています。

<?xml version="1.0" encoding="UTF-8"?>
<ResultSet>
  <List>
  <FileName>C:\...\address.xlsx</FileName>
  <StartRow>2</StartRow>
  <SheetName>Sheet1</SheetName>
  <SearchCol>3</SearchCol>
  <SearchText>府中市</SearchText>
  <ItemIndex>1</ItemIndex>
  <OutPut>outFile.xlsx</OutPut>
  </List>
</ResultSet>

各パラメーターの説明はこちらです。

項目	説明	例
FileName	元ファイルのパス	C:\…\address.xlsx
StartRow	データ開始行	2
SheetName	元ファイルのシート名	Sheet1
SearchCol	検索対象の列No.	3
SearchText	検索文字	府中市
ItemIndex	項目行No.	1
OutPut	出力ファイル名	outFile.xlsx

[ad#top-1]

ソースコード

ソースコードは以下の通りです。

package main

import (
  "fmt"
  "github.com/tealeg/xlsx"
  "io/ioutil"
  "encoding/xml"
)

type List struct {
  FileName string
  StartRow int
  SheetName string
  SearchCol int
  SearchText string
  ItemIndex int
  OutPut string
}

type ResultSet struct {
  List []List
}

var conf ResultSet

func getConf(){
  data, _ := ioutil.ReadFile("conf.xml")
  err := xml.Unmarshal(data, &amp;conf)
  if err != nil { panic(err) }
}

func main() {
  //パラメーターをxmlファイルから取得する
  getConf()

  fmt.Println("ファイル読み込み中...")

  //Excelファイルを開く
  excel, err1 := xlsx.OpenFile(conf.List[0].FileName)
  if err1 != nil { fmt.Printf(err1.Error()) }

   fmt.Println("指定シート検索中...")

  //指定シートを検索して変数sheetへ
  var sheet *xlsx.Sheet
  for _, s := range excel.Sheets {
    if s.Name == conf.List[0].SheetName{
      sheet = s
      break
    }
  }

  //パラメーター
  index := conf.List[0].SearchCol-1
  text := conf.List[0].SearchText
  itemIndex := conf.List[0].ItemIndex - 1
  startRow := conf.List[0].StartRow - 1

  //新しいExcelファイルを作成
  newfile := xlsx.NewFile()
  newsheet, err := newfile.AddSheet("Sheet1")
  if err != nil { fmt.Println(err.Error()) }

  fmt.Println("検索中...")

  //検索
  r := 1; cnt := 0
  for _, row := range sheet.Rows {
    //項目行を最初に追加
    if cnt==itemIndex {
      c := 0
      for _, cell := range row.Cells {
        text, _ := cell.String()
        newsheet.Cell(0, c).Value = text
        c++
      }
    }
    //検索ヒットした行を新ファイルに追加
    if cnt>=startRow && row.Cells[index].Value==text {
      c := 0
      for _, cell := range row.Cells {
        text, _ := cell.String()
        newsheet.Cell(r, c).Value = text
        c++
      }
      r++
    }
    cnt++
  }

  //ファイルに保存（ファイル名はパラメーターから取得）
  err = newfile.Save(conf.List[0].OutPut)
  if err != nil { fmt.Println(err.Error()) }

  fmt.Println("完了！")
}