一个读写excel的简单程序(golang)

最近总有一些临时统计的需求，比如其他团队生产的一批数据，需要确认这批数据是否入到数仓，提供的列表就是一个excel，我们就需要读取excel中的所有数据，之后查询数仓数据库确认这批数据是否存在，并分别将存在的与不存在的再标记出来。

1、读取excel文件

首先准备一个excel，内容如下，大概有100条数据：

下面实现对该文档的读取，读取方式包括一次性全部读取，适合数据量不大的场景，也包括按行读取，适合数据量大的场景：

func ExcelRead(name string) {f, err := excelize.OpenFile(name)if err != nil {fmt.Println("open file error:", err)return}defer f.Close()// 获取所有sheet的名称sheets := f.GetSheetMap()for _, name := range sheets {// 获取单个sheet的全部内容，正式上线需要考虑内存占用问题rows, err := f.GetRows(name)if err != nil {fmt.Println("get rows error:", err)return}fmt.Println("rows lenth:", len(rows))// 每次读取一行，减少内存占用for i := 0; i < 200; i++ {st, err := f.GetCellValue(name, fmt.Sprintf("A%v", i+2))if err != nil {fmt.Println("get cell value error:", err)}if st == "" {break}et, err := f.GetCellValue(name, fmt.Sprintf("B%v", i+2))if err != nil {fmt.Println("get cell value error:", err)}sub, err := f.GetCellValue(name, fmt.Sprintf("C%v", i+2))if err != nil {fmt.Println("get cell value error:", err)}fmt.Println(st, et, sub)}}
}

运行记录：

2、写excel文件

写excel，那就把第一步读取的excel重新写入到excel中，并写到两个sheet中，内容为三列(A/B/C)：


func ExcelWrite(sheet1 [][]string, sheet2 [][]string) {col := []string{"A", "B", "C"}f := excelize.NewFile()defer f.Close()// 创建名为车端挖掘数据的工作表_, err := f.NewSheet("车端挖掘数据")if err != nil {fmt.Println("new sheet error:", err)return}// 设置单元格的值。f.SetCellValue("车端挖掘数据", "A1", "start_time")f.SetCellValue("车端挖掘数据", "B1", "end_time")f.SetCellValue("车端挖掘数据", "C1", "subject")for i, v := range sheet1 {for k, v1 := range v {f.SetCellValue("车端挖掘数据", col[k]+fmt.Sprintf("%v", i+2), v1)}}// 创建名为车端挖掘数据的工作表sheetName := "云端挖掘数据"index, err := f.NewSheet("云端挖掘数据")if err != nil {fmt.Println("new sheet error:", err)return}// 设置默认sheet，打开表格时，默认显示的sheetf.SetActiveSheet(index)// 设置单元格的值。f.SetCellValue(sheetName, "A1", "start_time")f.SetCellValue(sheetName, "B1", "end_time")f.SetCellValue(sheetName, "C1", "subject")for i, v := range sheet1 {for k, v1 := range v {f.SetCellValue(sheetName, col[k]+fmt.Sprintf("%v", i+2), v1)}}// 将Excel另存为文件if err := f.SaveAs("/Users/liupeng/Downloads/test.xlsx"); err != nil {fmt.Println(err)}
}

运行结果：

3、在excel中生成柱状图

假设我们有这样一组数据：

我们希望将这组数据写入excel，并生成一个柱状图，类似这样：

代码如下：

func CreateChart() {f := excelize.NewFile()defer f.Close()contents := [][]any{{nil, "Apple", "Orange", "Pear"},{"Small", 2, 3, 3},{"Normal", 5, 2, 4},{"Large", 6, 7, 8},}for idx, row := range contents {cell, err := excelize.CoordinatesToCellName(1, idx+1)if err != nil {fmt.Println(err)return}f.SetSheetRow("Sheet1", cell, &row)}if err := f.AddChart("Sheet1", "E1", &excelize.Chart{Type: excelize.Col3DClustered,Series: []excelize.ChartSeries{{Name:       "Sheet1!$A$2",Categories: "Sheet1!$B$1:$D$1",Values:     "Sheet1!$B$2:$D$2",},{Name:       "Sheet1!$A$3",Categories: "Sheet1!$B$1:$D$1",Values:     "Sheet1!$B$3:$D$3",},{Name:       "Sheet1!$A$4",Categories: "Sheet1!$B$1:$D$1",Values:     "Sheet1!$B$4:$D$4",}},Title: []excelize.RichTextRun{{Text: "Fruit Chart",},},}); err != nil {fmt.Println(err)return}// Save spreadsheet by the given path.if err := f.SaveAs("/Users/liupeng/Downloads/Book1.xlsx"); err != nil {fmt.Println(err)}
}

以上就是三个基础的操作excel的方法，希望对大家有用。

欢迎投身技术的小伙伴们关注交流~~~~~~~~~~~~~~~~~~。