我尝试创建一个简单的函数,将切片值附加到创建的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”所在的同一列。
答案 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/复制