【Python编程】CSV文件的读写操作

seven_ • 2023年2月21日下午10:36 • 硬件开发 • 阅读 217

CSV文件（Comma-Separated Value，逗号分割值，有时候也称为字符分割值，因为分割值的字符也可以不是逗号），其文件以纯文本形式存储表格数据（数据和文本）。纯文本存储则表明该文件是一个字符序列，不含必须像二进制数字那样被解读的数据。CSV文件可由任意数目的记录组成，每行记录之间以回车符‘\n’进行分割。在Python中内置了CSV文件的处理模块，借助该模块我们可以很方便地进行CSV文件的读写操作。
Python3中CSV模块的相关使用说明：https://docs.python.org/3/library/csv.html
Python2中CSV模块的相关使用说明：https://docs.python.org/2/library/csv.html

import os
import sys
import csv
import argparse

def csvReader(path, row, line):
    with open(path, 'rt') as csvfile:
        lines = csv.reader(csvfile)
        line_store = list(lines)

        print(line_store)
        print(line_store[row][line])
        return line_store[row][line]

def csvWriter(path, row, line, value):
    with open(path, 'rt') as csvfile:
        lines = csv.reader(csvfile)
        line_store = list(lines)
        line_store[row][line] = value
    with open(path, 'wt') as csvfile:
        myFile = csv.writer(csvfile)
        myFile.writerows(line_store)


if __name__ == "__main__":
    parser = argparse.ArgumentParser(description = 'csv R/W test')
    parser.add_argument('--func', dest='func', type=str, default='', action='store')
    parser.add_argument('--value', dest='value',type=int, default=0, action='store')
    parser.add_argument('--path', dest='path', type=str, default='./test.csv', action='store')
    parser.add_argument('--row', dest='row', type=int, default=0, action='store')
    parser.add_argument('--line', dest='line',type=int, default=0, action='store')
    args = parser.parse_args()

    if args.func == 'W':
        csvWriter(args.path, args.row, args.line, args.value)
    elif args.func =='R':
        csvReader(args.path, args.row, args.line)