将切片附加到csv golang

时间:2013-07-13 10:48:24

标签: csv go

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

| columnA | columnB |

| 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”所在的同一列。

2 个答案:

答案 0 :(得分:3)

您的代码包含我无法理解的内容,因此这是一个小程序,可以满足您的要求。你可以建立它。

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)
    }
}

答案 1 :(得分:1)

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/复制