企业🤖AI Agent构建引擎,智能编排和调试,一键部署,支持私有化部署方案 广告
[TOC] go语言自带的有csv文件读取模块,看起来好像不错,今天玩玩,也算是系统学习go语言的一部分……^\_^ # 一、写csv文件 函数: func NewWriter(w io.Writer) \*Writer func (w \*Writer) Flush() func (w \*Writer) Write(record \[\]string) (err os.Error) func (w \*Writer) WriteAll(records \[\]\[\]string) (err os.Error) 具体见官网:[http://golang.org/pkg/csv/#NewWriter](http://golang.org/pkg/csv/#NewWriter) 其实写csv文件大可不必用这些函数,直接按照规则写入文件就行,不过这里有这些函数,还是写个demo,也方便我以后使用: ~~~ package main import ( "csv" "os" "bytes" "fmt" ) func main(){ fileName := "2.csv" buf := new(bytes.Buffer) r2 := csv.NewWriter(buf) for i:=0;i<10;i++ { s := make([]string,3) s[0] = "user id" s[1] = "name" s[2] = "depart" r2.Write(s) r2.Flush() } fmt.Println(buf) fout,err := os.Create(fileName) defer fout.Close() if err != nil { fmt.Println(fileName,err) return } fout.WriteString(buf.String()) } ~~~ # 二、读csv文件 函数: func NewReader(r io.Reader) \*Reader func (r \*Reader) Read() (record \[\]string, err os.Error) func (r \*Reader) ReadAll() (records \[\]\[\]string, err os.Error) 具体见官网:[http://golang.org/pkg/csv/#Reader.Read](http://golang.org/pkg/csv/#Reader.Read) 1、首先要解决的是读取文件的所有内容并存成字符串,这里用的是ioutil模块,具体代码如下: ~~~ package main import ( "io/ioutil" "fmt" ) func main(){ b,err := ioutil.ReadFile("1.csv") if err != nil { panic(err.String()) } //fmt.Println(b) fmt.Printf("%s",string(b)) } ~~~ 2、读取csv文件内容: ~~~ package main import ( "fmt" "csv" "strings" "io/ioutil" ) func main(){ //fileName := "1.csv" fmt.Printf("Input file name : ") var fileName string fmt.Scanf("%s",&fileName) cntb,err := ioutil.ReadFile(fileName) if err != nil { panic(err.String()) } r2 := csv.NewReader(strings.NewReader(string(cntb))) ss,_ := r2.ReadAll() //fmt.Println(ss) sz := len(ss) for i:=0;i<sz;i++{ fmt.Println(ss[i]) } } ~~~ ~~~ 好,就这些了,希望对你有帮助 ~~~