登录
首页 >  Golang >  Go问答

Golang中如何对字符串切片进行元素过滤操作?

来源:stackoverflow

时间:2024-02-16 23:45:22 160浏览 收藏

从现在开始,努力学习吧!本文《Golang中如何对字符串切片进行元素过滤操作?》主要讲解了等等相关知识点,我会在golang学习网中持续更新相关的系列文章,欢迎大家关注并积极留言建议。下面就先一起来看一下本篇正文内容吧,希望能帮到你!

问题内容

首先,我是新来的,我正在尝试学习 golang。我想检查我的 csv 文件(有 3 个值;类型、制造商、型号)并创建一个新文件,在过滤操作之后我想将新数据(过滤后)写入创建的 csv 文件。这是我的代码,以便您可以更清楚地理解我。

package main

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

func main() {
    //openning my csv file which is vehicles.csv
    recordFile, err := os.Open("vehicles.csv")
    if err != nil{
        fmt.Println("An error encountered ::", err)
    }
    //reading it
    reader := csv.NewReader(recordFile)
    vehicles, _ := reader.ReadAll()
    //creating a new csv file
    newRecordFile, err := os.Create("newCsvFile.csv")
    if err != nil{
        fmt.Println("An error encountered ::", err)
    }
    //writing vehicles.csv into the new csv
    writer := csv.NewWriter(newRecordFile)
    err = writer.WriteAll(vehicles)
    if err != nil {
        fmt.Println("An error encountered ::", err)
    }
}

我构建之后,它就是这样工作的。它将所有数据读取并写入新创建的 csv 文件。但这里的问题是,我想过滤读取的 csv 的重复项,即 车辆,我正在创建另一个函数(在主函数之外)来过滤重复项,但我不能这样做,因为 >vehicles 的类型是 [][]string,我在互联网上搜索了有关过滤重复项的信息,但我发现的都是 int 或 string 类型。我想要做的是创建一个函数并在 writeall 操作之前调用它,以便 writeall 可以将正确的(重复过滤的)数据写入新的 csv 文件。请帮帮我!! 我很感激任何答案。 快乐编码!


正确答案


这取决于您如何定义“唯一性”,但总的来说,这个问题有几个部分。

什么是独特的?

  1. 所有字段必须相同
  2. 只有部分字段必须相等
  3. 在比较之前对部分或所有字段进行标准化

您有几种方法可以发挥您的独特性,包括:

  1. 您可以使用由唯一性“片段”键控的地图,需要 o(n) 状态
  2. 您可以在迭代时对记录进行排序并与先前的记录进行比较,需要 o(1) 状态,但更复杂

您有两种过滤和输出的方法:

  1. 您可以使用循环基于旧切片构建一个新切片并一次写入所有切片,这需要 o(n) 空间
  2. 如果不需要排序,可以随时将记录写入文件,这需要 o(1) 空间

我认为一个相当简单且高性能的方法是从第一个中选择(1),从第二个中选择(1),从第三个中选择(2),它们一起看起来像:

package main

import (
    "encoding/csv"
    "errors"
    "io"
    "log"
    "os"
)

func main() {
    input, err := os.Open("vehicles.csv")
    if err != nil {
        log.Fatalf("opening input file: %s", err)
    }

    output, err := os.Create("vehicles_filtered.csv")
    if err != nil {
        log.Fatalf("creating output file: %s", err)
    }
    defer func() {
        // Ensure the file is closed at the end of the program
        if err := output.Close(); err != nil {
            log.Fatalf("finalizing output file: %s", err)
        }
    }()

    reader := csv.NewReader(input)
    writer := csv.NewWriter(output)

    seen := make(map[[3]string]bool)
    for {
        // Read in one record
        record, err := reader.Read()
        if errors.Is(err, io.EOF) {
            break
        }
        if err != nil {
            log.Fatalf("reading record: %s", err)
        }
        if len(record) != 3 {
            log.Printf("bad record %q", record)
            continue
        }

        // Check if the record has been seen before, skipping if so
        key := [3]string{record[0], record[1], record[2]}
        if seen[key] {
            continue
        }
        seen[key] = true

        // Write the record
        if err := writer.Write(record); err != nil {
            log.Fatalf("writing record %d: %s", len(seen), err)
        }
    }
}

以上就是《Golang中如何对字符串切片进行元素过滤操作?》的详细内容,更多关于的资料请关注golang学习网公众号!

声明:本文转载于:stackoverflow 如有侵犯,请联系study_golang@163.com删除
相关阅读
更多>
最新阅读
更多>
课程推荐
更多>