VB.NET实现按首列内容拆分单个CSV文件为多个CSV文件

原代码存在的问题 #

• 数组索引错误：VB.NET数组默认从0开始，原循环For x_DOF = 1 TO LineCount会跳过第一行，且当x_DOF等于LineCount时，访问LineOfMonitorID(x_DOF)会触发索引越界。
• LineToWrite赋值错误：内层循环中每次将LineToWrite设为当前列的值，最终只保留每行最后一列的内容，而非整行数据。
• 硬编码判断限制：仅判断101和111，无法处理其他第一列值，扩展性差。
• StreamWriter使用不当：每次循环都创建新的写入器，不仅效率低，还存在拼写错误（.scv应为.csv）；且未正确管理资源，可能导致文件未关闭或内容写入异常。
• 冗余操作与未初始化变量：重复读取文件（StreamReader和File.ReadAllLines），LineCount、2DArray、MonitorID等变量未声明或初始化，导致运行异常。

修正后的代码 #

以下是优化后的实现，支持自动按第一列所有值拆分文件，无需硬编码：

Imports System.IO
Imports System.Collections.Generic

Module SplitCsvByFirstColumn
    Sub Main()
        Dim mainFilePath As String = "R:\FilePath\MainFile.csv"
        Dim outputDir As String = "R:\PathToNewFile\"
        
        ' 创建输出目录（如果不存在）
        If Not Directory.Exists(outputDir) Then
            Directory.CreateDirectory(outputDir)
        End If
        
        ' 用字典管理每个第一列对应的StreamWriter，避免重复创建
        Dim writers As New Dictionary(Of String, StreamWriter)()
        
        Try
            ' 逐行读取主CSV文件
            For Each line As String In File.ReadAllLines(mainFilePath)
                ' 跳过空行
                If String.IsNullOrWhiteSpace(line) Then Continue For
                
                ' 拆分每行数据，注意：如果CSV中有带逗号的字段，这种拆分方式会有问题，需用专门的CSV解析库
                Dim splits As String() = line.Split(New Char() {","c}, 2) ' 只拆分第一列，避免字段内逗号影响
                Dim monitorId As String = splits(0).Trim() ' 去除首尾空格
                
                ' 获取或创建对应的StreamWriter
                If Not writers.ContainsKey(monitorId) Then
                    Dim outputPath As String = Path.Combine(outputDir, $"{monitorId}.csv")
                    writers.Add(monitorId, New StreamWriter(outputPath, True))
                End If
                
                ' 写入整行数据
                writers(monitorId).WriteLine(line)
            Next
        Finally
            ' 关闭所有StreamWriter，释放资源
            For Each writer In writers.Values
                writer.Close()
            Next
        End Try
    End Sub
End Module

关键优化点 #

• 使用字典管理写入器：每个第一列值对应一个StreamWriter，避免重复打开/关闭文件，提升效率。
• 灵活处理所有第一列值：无需硬编码判断，自动为每个不同的第一列值创建对应CSV文件。
• 正确处理整行写入：直接写入原行内容，避免拆分后拼接错误。
• 资源安全管理：用Try-Finally确保所有写入器都能被正确关闭，防止文件占用。
• 处理空行与空格：跳过空行，去除第一列的首尾空格，避免因空格导致的文件名错误。

注意：如果你的CSV文件中存在带逗号的字段（比如101,"Jacobs, Jr.",123456），直接用Split方法会拆分错误，这种情况建议使用专门的CSV解析库（如CsvHelper）来处理。

内容的提问来源于stack exchange，提问作者Nate