数据处理---2.CSV文件读写

处理CSV文件代码：

1.读取csv文件

    # 读取CSV文件
    def readCsv(self, filePath):
    	data = []
   		try:
           	print("------文件格式正确，开始读取文件------")
               for i in csv.reader(open(filePath, 'r')):
               	# print(i)
                   data.append(i)
          except EOFError:
          	   print("读取文件错误")
        return data

PS:type(data) is list

2.csv写入

    # CSV写入数据
    def writeCsv(self, filePath, write_data):
    """
    filePath:文件地址
    write_data:写入数据
    """
        try:
            print("------write_data-------")
            print(write_data)
            data_list = self.readCsv(filePath) + write_data
            f = open(path, 'w', newline='')
            writer = csv.writer(f)
            writer.writerow(data_list)
        except EOFError:
            print("写入文件错误")

3.csv文件存储地址获取

# 获取文件的相对路径
def getFilePath(dir_name=None, file_name=None):
	"""
	dir_name:文件相对地址路径
	file_name:文件名
	"""
    filePath = None
    if dir_name is None and file_name is None:
        filePath = os.path.dirname(os.path.dirname(os.path.realpath(__file__)))
    elif dir_name is not None and file_name is None:
        filePath = os.path.join(os.path.dirname(os.path.dirname(os.path.realpath(__file__))), dir_name)
    elif dir_name is not None and file_name is not None:
        filePath = os.path.join(os.path.join(os.path.dirname(os.path.dirname(os.path.realpath(__file__))), dir_name), file_name)
    else:
        print("异常情况，请检查逻辑")
    return filePath

os.path.dirname(os.path.dirname(os.path.realpath(file)))—项目文件根目录

4.判断文件夹下最新的文件是否为csv格式

 # 判断最新的文件是否是Csv格式
    def isCsv(self):
        if '.csv' in str(self.getFilePath()):
            print("File is Csv:", self.getFilePath())
            return True
        else:
            print("File is Not Csv:", self.getFilePath())
            return False