繁体   English   中英

将切片附加到csv golang

[英]append slice to csv golang

我试图创建一个简单的函数,将切片值附加到创建的csv文件的下一列。 目的是在test.csv中具有这种样式

|列A |列B |

| header1 | header2 |

| FIRSTVALUE | SECONDVALUE |

(很抱歉,格式错误,但是一旦生成了text.csv,如果我们打开text.csv文件,我们的想法是将“ SECONDVALUE”放在“ FIRSTVALUE”之后的下一列中(如并在每次调用该函数时,都将在下一行的前一列之后追加下一个值)。

这就是我得到的结果,但是它直接将SECONDVALUE附加到FIRSTVALUE的同一列上。(所以不是得到“ | FIRSTVALUE | SECONDVALUE”而是得到“ | FIRSTVALUESECONDVALUE |”)

package main

import (
    "encoding/csv"
    "fmt"
    "os"
    "strings"
)


func main() {
    callfunc()
}

func callfunc() int {

    testval1 := []string{"FIRST VALUE"}
    test(testval1, "test.csv", 1)

    testval2 := []string{"SECOND VALUE"}
    test(testval2, "test.csv", 0) //I want this to be added to the column next to first value 

    return 1
}


func test(lines []string, path string, header int) {

    var hdr = []string{"header1", "header2"}

    fmt.Println("inside the method..\n")

    file, err := os.OpenFile(path, os.O_APPEND|os.O_WRONLY, 0600)
    if err != nil {
        if file, err = os.Create(path); err != nil {
            return
        }
    }
    defer file.Close()

    writer := csv.NewWriter(file)

    if header == 1 {
        returnError := writer.Write(hdr)
        if returnError != nil {
            fmt.Println(returnError)
        }
    }

    writer.Flush()

    for _, value := range lines {
        _, err := file.WriteString(strings.TrimSpace(value))
        if err != nil { //exception handler
            fmt.Println(err)
            break
        }
    }

    writer.Flush()
}

如何修复上面的代码行,以使“ SECONDVALUE”是第二列,而不是在“ FIRSTVALUE”所在的列中。

您的代码包含我无法清楚理解的内容,因此这是一个微型程序,可以满足您的要求。 您可以在此基础上构建。

package main

import (
    "encoding/csv"
    "os"
)

func addcol(fname string, column []string) error {
    // read the file
    f, err := os.Open(fname)
    if err != nil {
        return err
    }
    r := csv.NewReader(f)
    lines, err := r.ReadAll()
    if err != nil {
        return err
    }
    if err = f.Close(); err != nil {
        return err
    }

    // add column
    l := len(lines)
    if len(column) < l {
        l = len(column)
    }
    for i := 0; i < l; i++ {
        lines[i] = append(lines[i], column[i])
    }

    // write the file
    f, err = os.Create(fname)
    if err != nil {
        return err
    }
    w := csv.NewWriter(f)
    if err = w.WriteAll(lines); err != nil {
        f.Close()
        return err
    }
    return f.Close()
}

func main() {
    col := []string{"column two", "a", "b", "c", "d"}
    if err := addcol("in.csv", col); err != nil {
        panic(err)
    }
}
func addcol(fname string, column []string)  {
 // read the file
 f, err := os.OpenFile(fname, os.O_WRONLY|os.O_CREATE|os.O_APPEND, 0644)
 if err != nil {
     fmt.Println("Error: ", err)
     return
 }
 w := csv.NewWriter(f)
 w.Write(column)
 w.Flush()
}

https://www.reddit.com/r/golang/comments/46fdef/help_appending_a_slice_to_next_row_in_csv_file/复制

暂无
暂无

声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.

 
粤ICP备18138465号  © 2020-2024 STACKOOM.COM