原创C#CSV读写类 郝伟 2021/01/14 [TOC]

1. 简介

网上找了一些C#的第三方的CSV库总觉得不好用,于是自己写了一个 CSV 类,用于处理CSV函数。

2. 主要特性

  • 使用简单 只需要几行代码即可定义、访问和进行相应的读写操作。
  • 功能完善 包括CSV操作的主要功能。
  • 支持不同编码 可以自定义,并支持所有常见编码格式。
  • 支持特殊字符 如引号("),逗号(,), 换行符号(\r\n)。
  • 只有一个源文件 源代码只有1个,方便管理。

3. 主要用法

/***************** 从文件初始化一个CSV文件 *****************/
// 初始化,可修改编码,如:new CSV(input, Encoding.Default);  
// 加载后的数据在 csv.Data 中,其数据类型为 List<List<string>>
CSV csv = new CSV(@"C:\input\data.csv");  

/***************** 从加载好的 csv.Data 中读取数据 *****************/
string v0_0 = csv.Data[0,0]
// 为了简化操作,可以用以下方式读取。注意:这两种方法都会对范围验证,如果数据不存在,则返回 null
string v3_5 = csv.GetData(3,5);    // 将单元格(3,5)的内容为赋给 v3_5
string c0_1 = csv[0,1];            // 将单元格(0,1)的内容为赋给 c3_5    

/***************** 从加载好的 csv.Data 中读取数据 *****************/
csv.SetData(2,6,"tom");         // 设置单元格(2,6)的内容为 "tom"
csv[1, 5] = "jack";             // 设置单元格(1,5)的内容为 "jack"

/***************** 添加一行数据 *****************/
// 向csv.Data追尾追加一行数据,长度为 ColumnCount,每个元素均为 ""。
csv.AddRow();                  
// 向csv.Data追加一行数据,如果 csv.ColumnCount 长度大于2,则不足部分用""补充。   
csv.AddRow(new String[]{"a", "b"}); 
// 向csv.Data的第5行追加一行数据,如果 csv.ColumnCount 长度大于2,则不足部分用""补充。  
// 如果行号小于0,则在首行添加;如果大于最大长度则在末尾添加。 
csv.AddRow(new String[]{"a", "b"}, 5); 

/***************** 删除列数据 *****************/
// 删除第11列数据,如果超出列的范围,则不做任何操作。
csv.RemoveColumn(10) 
// 删除第1,3,8列数据。当有重复的列时,只删除一次。范围超出的不做操作。
csv.RemoveColumns(1, 3, 8, 3)

/***************** 保存数据 *****************/
csv.Save("c:\\output\\data1.csv");

4. 源代码

using System;
using System.Collections.Generic;
using System.IO;
using System.Linq;
using System.Text;

namespace NMapMatchingML
{
    /// <summary>
    /// 用于CSV文件的读写,加载以后数据以二维表的形式返回。
    /// 主要用法如下:
    /// <example>  
    /// CSV csv = new CSV("C:\\input\\data.csv");  // 初始化    
    /// string c0_1 = csv[0,1];         // 读取 
    /// csv[1, 5] = "jack";             // 写入 
    /// csv.Save("c:\\output\\data1.csv");          // 保存 
    /// </example>
    /// </summary>
    public class CSV
    {
        /// <summary>
        /// 文件的编码。
        /// </summary>
        public Encoding CsvEncoding { get; set; } = Encoding.UTF8;

        /// <summary>
        /// 表示是否要进行列的裁剪操作。
        /// </summary>
        public bool TrimColumns { get; set; } = false;

        /// <summary>
        /// CSV的分隔符,默认为半角逗号(即,)。
        /// </summary>
        public char CsvSeparator { get; set; } = ',';

        /// <summary>
        /// 所有的数据都保存在些数组中。
        /// </summary>
        public List<List<string>> Data { get; set; } = new List<List<string>>();

        /// <summary>
        /// CSV是一种通用的使用逗号隔开的二维表的结构文件。
        /// </summary>
        public CSV() { }

        /// <summary>
        /// 通过指定的CSV文件进行加载。返回一个二维矩阵。
        /// </summary>
        /// <param name="filepath"></param>
        public CSV(string filepath) : this(filepath, Encoding.Default) { }

        /// <summary>
        /// 通过指定的CSV文件进行加载。
        /// </summary>
        /// <param name="filepath">指定的CSV文件路径。</param>
        /// <param name="encoding">CSV文本的编码。</param>
        public CSV(string filepath, Encoding encoding)
        {
            CsvEncoding = encoding;
            Read(filepath);
        }

        /// <summary>
        /// 返回CSV二维数列的总行数。
        /// </summary>
        /// <returns></returns>
        public int RowCount { get { return Data.Count; } }

        /// <summary>
        /// 返回CSV二维数列的总列数。
        /// </summary>
        public int ColumnCount { get { return Data != null && Data.Count > 0 ? Data[0].Count : 0; } }

        /// <summary>
        /// 返回指定行列的值,如果超出数据返回值为 null。
        /// </summary>
        /// <param name="row">指定行。</param>
        /// <param name="column">指定列。</param>
        /// <returns></returns>
        public string GetData(int row, int column)
        {
            return IsInRange(row, column) ? Data[row][column] : null;
        }

        /// <summary>
        /// 读写CSV指定单元格,下标从0开始。如果不在范围内,则会操作失败,处理方式如下:
        /// 读=返回null;写=返回false。
        /// </summary>
        /// <param name="row">指定的行。</param>
        /// <param name="column">指定的列。</param>
        /// <returns></returns>
        public string this[int row, int column]
        {
            get { return GetData(row, column); }
            set { SetData(row, column, value); }
        }

        /// <summary>
        /// 在指定行添加一行数据内容为""的字符数列。
        /// 输入数据的行数必需与 RowCount 相同,超出部分被裁掉,不足部分用""补充。
        /// 要插入的行的位置小于0的按0处理,大于等于RowCount的按RowCount处理。
        /// </summary>
        /// <param name="columnData">输入数据。</param>
        /// <param name="columnID">需要添加的行的序号。</param>
        public void AddColumn(string[] columnData = null, int columnID = int.MaxValue)
        {
            // 输入数据的长度必需与 ColumnCount相同,超出部分被裁掉,不足部分用""补充。
            columnData = columnData ?? new string[RowCount];
            List<string> list = new List<string>();
            for (int i = 0; i < ColumnCount; i++)
                list.Add(i < columnData.Length ? columnData[i] : "");

            // 要插入的行的位置小于0的按0处理,大于等于ColumnCount的按ColumnCount处理。
            columnID = columnID < 0 ? 0 : columnID;
            columnID = columnID >= ColumnCount ? ColumnCount : columnID;

            // 将数据插入指定位置。
            for (int i = 0; i < RowCount; i++)
                Data[i].Insert(columnID, columnData[i]);

        }

        /// <summary>
        /// 删除指定的某个列。
        /// </summary>
        /// <param name="columnId">待删除列的ID,范围为[0, ColumnCount)。</param>
        public void RemoveColumn(int columnId)
        {
            if (0 <= columnId && columnId < ColumnCount)
            {
                for (int row = 0; row < RowCount; row++)
                {
                    Data[row].RemoveAt(columnId);
                }
            }
        }

        /// <summary>
        /// 删除指定的多个列。
        /// </summary>
        /// <param name="columnIds">待删除列的列的编号,可以是多个列。如果不在范围内,不会处理;如果有重复,只会处理1次。</param>
        public void RemoveColumns(params int[] columnIds)
        {
            columnIds.Distinct().OrderByDescending(c => c).ToList().ForEach(c => RemoveColumn(c));
        }


        /// <summary>
        /// 在指定行添加一行数据内容为""的字符数列。
        /// 输入数据的长度必需与 ColumnCount相同,超出部分被裁掉,不足部分用""补充。
        /// 要插入的行的位置小于0的按0处理,大于等于RowCount的按RowCount处理。
        /// </summary>
        /// <param name="strs">输入数据。</param>
        /// <param name="rowID">需要添加的行的序号。</param>
        public void AddRow(string[] strs = null, int rowID = int.MaxValue)
        {
            // 输入数据的长度必需与 ColumnCount相同,超出部分被裁掉,不足部分用""补充。
            strs = strs ?? new string[ColumnCount];
            List<string> list = new List<string>();
            for (int i = 0; i < ColumnCount; i++)
                list.Add(i < strs.Length ? strs[i] : "");

            // 要插入的行的位置小于0的按0处理,大于等于RowCount的按RowCount处理。
            rowID = rowID < 0 ? 0 : rowID;
            rowID = rowID >= RowCount ? RowCount : rowID;
            Data.Insert(rowID, list);
        }

        /// <summary>
        /// 在指定行添加一行数据内容为""的字符数列。
        /// </summary>
        /// <param name="rowNo">待插入的行号。</param>
        /// <param name="rowdata">待插入的数据。</param>
        public void AddRow(int rowNo, params string[] rowdata)
        {
            AddRow(rowdata, rowNo);
        }

        /// <summary>
        /// 判断指定行列的单元格是否在范围内。
        /// </summary>
        /// <param name="row">指定的行编号。</param>
        /// <param name="column">指定的列编号。</param>
        /// <returns></returns>
        public bool IsInRange(int row, int column) { return row >= 0 && row < RowCount && column >= 0 && column < ColumnCount; }

        /// <summary>
        /// 为指定编号的行列的单元格赋值。
        /// </summary>
        /// <param name="row">所要设置的行。</param>
        /// <param name="column">所要设置的列。</param>
        /// <param name="value">所要设置的值。</param>
        /// <returns></returns>
        public bool SetData(int row, int column, string value)
        {
            bool valid = IsInRange(row, column);
            if (valid)
                Data[row][column] = value;
            return valid;
        }

        /// <summary>
        /// 读取CSV数据文件, 目前的版本不支持带有分隔符的内容。
        /// </summary>
        /// <param name="csvfile">待加载的CSV文件。</param>
        public void Read(string csvfile)
        {
            Data = GetListCsvData(csvfile);
        }

        /// <summary>
        /// 将数据转换为二维列表。
        /// </summary>
        /// <returns></returns>
        public List<List<string>> GetListCsvData(string file)
        {
            StreamReader reader = new StreamReader(file, Encoding.UTF8);
            List<List<string>> tempListCsvData = new List<List<string>>();
            bool isNotEndLine = false;
            string tempCsvRowString = reader.ReadLine();

            // 对每行进行读写
            while (tempCsvRowString != null)
            {
                List<string> tempCsvRowList;
                if (isNotEndLine)
                {
                    tempCsvRowList = ParseContinueLine(tempCsvRowString);
                    isNotEndLine = (tempCsvRowList.Count > 0 && tempCsvRowList[tempCsvRowList.Count - 1].EndsWith("\r\n"));
                    List<string> myNowContinueList = tempListCsvData[tempListCsvData.Count - 1];
                    myNowContinueList[myNowContinueList.Count - 1] += tempCsvRowList[0];
                    tempCsvRowList.RemoveAt(0);
                    myNowContinueList.AddRange(tempCsvRowList);
                }
                else
                {
                    tempCsvRowList = ParseLine(tempCsvRowString);
                    isNotEndLine = (tempCsvRowList.Count > 0 && tempCsvRowList[tempCsvRowList.Count - 1].EndsWith("\r\n"));
                    tempListCsvData.Add(tempCsvRowList);
                }
                tempCsvRowString = reader.ReadLine();
            }
            reader.Close();

            // 读取完成以后,可能不同的行有不同的数据个数,为了保证所有行的内容一样,添加以下内容:
            // 找到最大列数。
            int maxColumn = 0;
            for (int i = 0; i < tempListCsvData.Count; i++)
                maxColumn = tempListCsvData[i].Count > maxColumn ? tempListCsvData[i].Count : maxColumn;

            // 使所有行的列数都是 maxColumn
            foreach (List<string> item in tempListCsvData)
                while (item.Count < maxColumn)
                    item.Add("");

            return tempListCsvData;
        }


        /// <summary>
        /// 处理未完成的Csv单行
        /// </summary>
        /// <param name="line">数据源</param>
        /// <returns>元素列表</returns>
        private List<string> ParseContinueLine(string line)
        {
            StringBuilder _columnBuilder = new StringBuilder();
            List<string> Fields = new List<string>();
            _columnBuilder.Remove(0, _columnBuilder.Length);
            if (line == "")
            {
                Fields.Add("\r\n");
                return Fields;
            }

            for (int i = 0; i < line.Length; i++)
            {
                char character = line[i];

                if ((i + 1) == line.Length)//这个字符已经结束了整行
                {
                    if (character == '"') //正常转义结束,且该行已经结束
                    {
                        Fields.Add(TrimColumns ? _columnBuilder.ToString().TrimEnd() : _columnBuilder.ToString());
                        return Fields;
                    }
                    else //异常结束,转义未收尾
                    {
                        _columnBuilder.Append("\r\n");
                        Fields.Add(_columnBuilder.ToString());
                        return Fields;
                    }
                }
                else if (character == '"' && line[i + 1] == CsvSeparator) //结束转义,且后面有可能还有数据
                {
                    Fields.Add(TrimColumns ? _columnBuilder.ToString().TrimEnd() : _columnBuilder.ToString());
                    i++; //跳过下一个字符
                    Fields.AddRange(ParseLine(line.Remove(0, i + 1)));
                    break;
                }
                else if (character == '"' && line[i + 1] == '"') //双引号转义
                {
                    i++; //跳过下一个字符
                }
                else if (character == '"') //双引号单独出现(这种情况实际上已经是格式错误,为了兼容暂时不处理)
                {
                    throw new Exception("格式错误,错误的双引号转义");
                }
                _columnBuilder.Append(character);
            }
            return Fields;
        }

        /// <summary>
        /// 解析一行CSV数据。
        /// </summary>
        /// <param name="line">待分析的行。</param>
        /// <returns></returns>
        private List<string> ParseLine(string line)
        {
            StringBuilder sb = new StringBuilder();
            List<string> Fields = new List<string>();
            bool inColumn = false;  //是否是在一个列元素里
            bool inQuotes = false;  //是否需要转义
            bool isNotEnd = false;  //读取完毕未结束转义
            sb.Remove(0, sb.Length);

            //空行也是一个空元素,一个逗号是2个空元素
            if (line == "")
                Fields.Add("");

            // 遍历一个行中的所有元素。
            for (int i = 0; i < line.Length; i++)
            {
                char c = line[i];
                if (!inColumn)
                {
                    inColumn = true;
                    if (c == '"')
                    {
                        inQuotes = true;
                        continue;
                    }
                }

                // 如果有引用 
                if (inQuotes)
                {
                    //这个字符已经结束了整行
                    if ((i + 1) == line.Length)
                    {
                        //正常转义结束,且该行已经结束
                        if (c == '"')
                        {
                            inQuotes = false;
                            continue;
                        }
                        isNotEnd = true;
                    }
                    //结束转义,且后面有可能还有数据
                    else if (c == '"' && line[i + 1] == CsvSeparator)
                    {
                        inQuotes = false;
                        inColumn = false;
                        i++;
                    }
                    //双引号转义
                    else if (c == '"' && line[i + 1] == '"')
                    {
                        i++;
                    }
                    //双引号单独出现(这种情况实际上已经是格式错误,为了兼容可暂时不处理)
                    else if (c == '"')
                    {
                        throw new Exception("格式错误,错误的双引号转义");
                    }
                }
                else if (c == CsvSeparator)
                {
                    inColumn = false;
                }

                //结束该元素时inColumn置为false,并且不处理当前字符,直接进行Add
                if (!inColumn)
                {
                    Fields.Add(TrimColumns ? sb.ToString().Trim() : sb.ToString());
                    sb.Remove(0, sb.Length);
                }
                else
                {
                    sb.Append(c);
                }
            }

            // 标准格式一行结尾不需要逗号结尾,而上面for是遇到逗号才添加的,为了兼容最后还要添加一次
            if (inColumn)
            {
                if (isNotEnd)
                    sb.Append("\r\n");

                Fields.Add(TrimColumns ? sb.ToString().Trim() : sb.ToString());
            }
            //如果inColumn为false,说明已经添加,因为最后一个字符为分隔符,所以后面要加上一个空元素
            else
            {
                Fields.Add("");
            }

            return Fields;
        }

        /// <summary>
        /// 将数据进行保存,默认使用编号进行保存。 
        /// </summary>
        /// <param name="outcsvfile">需要保存的路径。</param>
        public bool Save(string outcsvfile)
        {
            if (!File.Exists(outcsvfile))
                throw new Exception("find error in your FilePath");

            if (Data == null)
                throw new Exception("your DataSouse is null");

            // 将Data中的数据以非追加的方式写至指定的文件中。
            using (StreamWriter sw = new StreamWriter(outcsvfile, false, CsvEncoding))
            {
                foreach (List<string> fields in Data)
                {
                    StringBuilder sb = new StringBuilder();
                    for (int i = 0; i < fields.Count; i++)
                    {
                        sb.Append(fields[i].Contains("\"") ?
                            string.Format("\"{0}\"", fields[i].Replace("\"", "\"\"")) :
                            string.Format("\"{0}\"", fields[i]));

                        if (i < fields.Count - 1)
                            sb.Append(CsvSeparator);
                    }
                    sw.WriteLine(sb.ToString());
                }
            }

            return true;
        }

        /// <summary>
        /// 以二维表的形式返回氖Data的内容,同一行的相邻数据使用\t\t分隔。
        /// </summary>
        /// <returns></returns>
        public override string ToString()
        {
            StringBuilder sb = new StringBuilder();
            for (int i = 0; i < Data.Count; i++)
            {
                sb.AppendLine(string.Join("\t\t", Data[i]));
            }
            return sb.ToString();
        }
    }
}

results matching ""

    No results matching ""