Python的听课笔记案例8--空气污染指数计算4.0

2.0与3.0视频展示了如何使用它。Python处理JSON文件和CSV文件
这一部分扩展到让程序自动确定读取的文件是什么。JSON文件还是CSV归档，并进行相应的操作。

以下为补充内容：

1、CSV文件读取
import csv
csv.reader()以列表形式返回每行记录

2、使用with语句操作文件对象。
with open(file_name) as somefile:
for line in somefile:
print(line)
使用with语句，无论在文件处理过程中是否发生异常，都可以保证。with在执行语句后关闭文件，不需要close()语句。
使用with语句取代了打开文件、操作文件和关闭文件的操作。

3、os模块
os该模块提供系统和目录操作相关的功能，不受平台限制。

函数

含义

os.remove()

删除文件

os.makedirs()

创建多层目录

os.rmdir()

删除单级目录

os.rename()

重命名文件

os.path.isfile()

确定它是否为文件

os.path_isdir()

确定它是否为目录

os.path.join()

连接目录，如path1连接path2为path1/path2

os.path.splitext()

将文件拆分为文件名和扩展名，如Splittmp.txt为tmp和.txt

以下是代码：
1，主要功能是通过os模块的os.path.splitext()确定扩展采用哪个分支功能；
2、新建json解码功能；
3、新建csv解码功能。

"""
    作者：lanxingbudui
    日期：2020-02-23
    功能：AQI计算
    版本：4.0
"""

import json
import csv
import os

def process_json_file(file_path):
    """
        解码json文件
    """
    """
    处理单据的三个步骤：1，打开文件；2，处理文件；；3，关闭文件
    """
    # f = open(file_path, mode=r, encoding=utf-8)
    # city_list = json.load(f)
    # return city_list

    with open(file_path, mode=r, encoding=utf-8) as f:
        city_list = json.load(f)
    print(city_list)

def process_csv_file(file_path):
    """
        解码csv文件
    """
    with open(file_path, mode=r, encoding=utf-8,newline=) as f:
        reader = csv.reader(f)
        for row in reader:
            print(, .join(row))

def main():
    """
        主函数
    """
    file_path = input(请输入文件名：)
    file_name, file_ext = os.path.splitext(file_path)
    if file_ext == .json:
        # json文件
        process_json_file(file_path)
    elif file_ext == .csv:
        # csv文件
        process_csv_file(file_path)
    else:
        print(不支持的文件格式！)

if __name__ == __main__:
    main()

效果图