时间戳

首页 > 解决方案 > 使用 Go 从 csv 数据框的列创建文本文件

go - 使用 Go 从 csv 数据框的列创建文本文件

问题描述

我正在尝试遍历csv文件并输出标题为第一列中每一行的文本文件。然后用该列的其他行中的数据填充每个文本文件。我能够将内容打印csv到文本文件中，但是我无法使用 for 循环来获取逻辑以获取第一列的索引并使用它来创建/命名一个新.txt文件。

package main

import (
    "fmt"
    "io"
    "io/ioutil"
    "log"
    "os"
)

func main() {
    fmt.Println("Enter file path to CSV: ")
    var csvFile string
    _, err := fmt.Scanln(&csvFile)
    if err != nil {
        log.Fatal("Cannot read input")
        return
    }
    //open file
    inFile, err := os.Open(csvFile)
    if err != nil {
        log.Fatal(err)
    }
    defer inFile.Close()

    readMe, _ := ioutil.ReadAll(inFile)

    blankFile, err := os.Create(`C:\temp\test.txt`)
    if err != nil {
        log.Fatal(err)
    }
    defer blankFile.Close()

    //write data to text file
    outFile, err := blankFile.Write(readMe)
    if err == io.EOF {
        log.Fatalln("Failed")
    } else if err != nil {
        log.Fatal(err)
    }
    //print bytes total
    fmt.Println(outFile, " bytes printed")

}

标签： go

解决方案

从 csv 中获取多列并将每一列打印到一个新的文本文件中。

循环一个 csv 并生成一个新的文本文件，该文件将在第 1 行中的每一列之后命名。然后将使用该列的其他行中的数据填充每个文本文件。

例如，

package main

import (
    "encoding/csv"
    "fmt"
    "io"
    "os"
    "path/filepath"
)

func CsvFileToTxtFiles(inFile string) error {
    in, err := os.Open(inFile)
    if err != nil {
        return err
    }
    defer in.Close()
    r := csv.NewReader(in)
    hdr, err := r.Read()
    if err != nil {
        return err
    }
    f := make([]*os.File, len(hdr))
    w := make([]*csv.Writer, len(hdr))
    pfx := filepath.Clean(inFile)
    pfx = pfx[:len(pfx)-len(filepath.Ext(pfx))]
    for i, col := range hdr {
        var err error
        f[i], err = os.Create(pfx + "." + col + ".txt")
        if err != nil {
            return err
        }
        defer f[i].Close()
        w[i] = csv.NewWriter(f[i])
        if err != nil {
            return err
        }
        defer w[i].Flush()
    }

    for {
        row, err := r.Read()
        if err != nil {
            if row == nil && err == io.EOF {
                break
            }
            return err
        }
        for i, col := range row {
            err := w[i].Write([]string{col})
            if err != nil {
                return err
            }
        }
    }

    for i := range hdr {
        var err error
        w[i].Flush()
        err = w[i].Error()
        if err != nil {
            return err
        }
        err = f[i].Close()
        if err != nil {
            return err
        }
    }

    return nil
}

func main() {
    if len(os.Args) <= 1 {
        usage := "usage: " + filepath.Base(os.Args[0]) + " FILE"
        fmt.Fprintln(os.Stderr, usage)
        return
    }
    inFile := os.Args[1]
    err := CsvFileToTxtFiles(inFile)
    if err != nil {
        fmt.Fprintln(os.Stderr, err)
        return
    }
}

输出：

$ cat ioj.test.csv
one,two,three
1,2,3
11,22,33
$ go run ioj.go ioj.test.csv
$ cat ioj.test.one.txt
1
11
$ cat ioj.test.two.txt
2
22
$ cat ioj.test.three.txt
3
33
$

推荐阅读