csv - 将 slice 附加到csv golang

我试图创建一个简单的函数，将 slice 值附加到已创建的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)
    }
}