zoukankan      html  css  js  c++  java
  • golang csv问题

    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

    其实写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

    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])
            }
    }
  • 相关阅读:
    要开学了,暂时停更
    day13 IP包头分析 | 路由器原理 1
    day12 数据链路层 | 交换机基本命令
    day11 OSI与TCP-IP 5层协议 | 物理层相关知识
    day10 扫描与爆破
    day 09 简单渗透测试
    day07 PKI
    day07 域
    day06 WEB服务器 | FTP服务器
    day05 DHCP部署与安全 | DNS部署与安全
  • 原文地址:https://www.cnblogs.com/zhangym/p/6142738.html
Copyright © 2011-2022 走看看