Go来合并两个csv

简介: Go来合并两个csv

背景


通常我们处理文件都是用python+pandas,确实很香。但是今天突然有人找我用go写一个合并两个csv文件的,需要将两个csv按id进行合并,大致如下图


image.png

image.png

思路很简单,先读入两个csv文件,然后遍历比较id,如果相同就把另一个文件的内容加到第一个中。然后创建merge.csv,将上一步得到的数据写入csv即可。


代码部分


package main
import (
  "encoding/csv"
  "log"
  "os"
  "fmt"
)
func join(fileName1 string, fileName2 string) (row [][]string) {
  fs1, _ := os.Open(fileName1)
  r1 := csv.NewReader(fs1)
  content1, err := r1.ReadAll()
  if err != nil {
    log.Fatalf("can not readall, err is %+v", err)
  }
  fs2, _ := os.Open(fileName2)
  r2 := csv.NewReader(fs2)
  content2, err := r2.ReadAll()
  if err != nil {
    log.Fatalf("can not readall, err is %+v", err)
  }
  row = make([][]string, len(content1[1:]))
  for i, row1 := range content1[1:] {
    for _, row2 := range content2[1:] {
      if row1[0] == row2[0] {
        row[i] = append(row[i], row1[0], row1[1], row1[2], row2[1], row2[2])
      }
    }
  }
  return
}
func edit(filename string, row [][]string) {
  f, err := os.Create(filename)
  if err != nil {
    panic(err)
  }
  defer f.Close()
  //防止乱码
  f.WriteString("\xEF\xBB\xBF")
  w := csv.NewWriter(f)
  w.Write([]string{"ID", "name", "gender", "course", "score"})
  w.WriteAll(row)
  w.Flush()
}
func main() {
  row := join("1.csv", "2.csv")
  fmt.Println(row)
  edit("join.csv", row)
  fmt.Println("合并已完成!")
}
复制代码


image.png

image.png

优化


对于小数据量这样是没有问题,但是一旦数据量增大,不仅读取的内存占用增加,而且遍历比较会很消耗时间。所以优化的话我们可以以行读取,并且想办法使用go的goroutine来增加速度。唉,太久没怎么写go了,一开始切片都有点忘了,最近准备写一些关于go的内容,所以开始从头复习一遍go相关的知识,后期会攒起来慢慢发。

目录
相关文章
|
Go
Go学习笔记(三) 学会使用GO标准库,以CSV库为例
CSV读取 标准库提供了一系列实用函数的API,让我们避免重复造轮子. 为了学会使用一个已有的轮子,我们需要学看文档。Go所有标准库文档都放在https://golang.org/pkg/, 假如我们想读取一个CSV文件, 我们就需要学习csv包的用法。
1240 0
|
20天前
|
存储 Go 索引
go语言中数组和切片
go语言中数组和切片
33 7
|
20天前
|
Go 开发工具
百炼-千问模型通过openai接口构建assistant 等 go语言
由于阿里百炼平台通义千问大模型没有完善的go语言兼容openapi示例,并且官方答复assistant是不兼容openapi sdk的。 实际使用中发现是能够支持的,所以自己写了一个demo test示例,给大家做一个参考。
|
20天前
|
程序员 Go
go语言中结构体(Struct)
go语言中结构体(Struct)
93 71
|
19天前
|
存储 Go 索引
go语言中的数组(Array)
go语言中的数组(Array)
100 67
|
22天前
|
Go 索引
go语言for遍历数组或切片
go语言for遍历数组或切片
91 62
|
24天前
|
并行计算 安全 Go
Go语言中的并发编程:掌握goroutines和channels####
本文深入探讨了Go语言中并发编程的核心概念——goroutine和channel。不同于传统的线程模型,Go通过轻量级的goroutine和通信机制channel,实现了高效的并发处理。我们将从基础概念开始,逐步深入到实际应用案例,揭示如何在Go语言中优雅地实现并发控制和数据同步。 ####
|
20天前
|
存储 Go
go语言中映射
go语言中映射
32 11
|
22天前
|
Go
go语言for遍历映射(map)
go语言for遍历映射(map)
31 12
|
21天前
|
Go 索引
go语言使用索引遍历
go语言使用索引遍历
29 9