51Testing软件测试论坛

标题: Python CSV、 JSON 格式高级处理（下） [打印本页]

作者: lsekfe 时间: 2023-6-9 11:24
标题: Python CSV、 JSON 格式高级处理（下）
分隔符和编码
　　了解不同分隔符和编码的应用
　　接上篇CSV（Comma Separated Values）和JSON（JavaScript Object Notation）是常见的数据交换格式。CSV文件是以逗号作为字段之间的分隔符，每行表示一个记录，每个字段可以通过引号来进行引用。而JSON文件则采用键值对的方式来表示数据，每个键值对之间使用逗号进行分隔，多个键值对组合成一个对象。
　　在实际应用过程中，我们可能遇到各种不同的分隔符和编码的CSV和JSON文件，例如分隔符可以是制表符、空格或其他字符，编码可以是UTF-8、GBK等等。因此需要了解并正确识别这些分隔符和编码，才能准确地读取和处理这些文件。
　　如何识别和处理不同编码的 CSV 和 JSON 文件
　　Python提供了一些内置库来帮助我们读取和处理CSV和JSON文件。其中，csv模块用于读写CSV文件，json模块用于读写JSON文件。
　　CSV文件的读取和处理
　　在使用csv模块读取CSV文件时，需要指定文件的编码和分隔符，通常情况下默认的编码为utf-8，分隔符为逗号。如果需要使用其他编码和分隔符，可以通过设置相应的参数来实现。下面是一个读取CSV文件的示例代码：
　　import csv
　　with open('data.csv', 'r', encoding='utf-8') as f:
　　 reader = csv.reader(f, delimiter=',')
　　 for row in reader:
　　 print(row)

　　在这个示例中，我们使用了csv.reader()函数来打开并读取文件，其中delimiter参数指定了分隔符为逗号。如果需要使用其他分隔符，只需将该参数设置为相应的值即可。
　　JSON文件的读取和处理
　　在使用json模块读取JSON文件时，需要注意文件的编码格式。通常情况下，JSON文件采用UTF-8编码，可以直接使用json.load()函数从文件中读取数据。如果使用其他编码格式，则需要先将文件内容解码为UTF-8编码后再进行操作。
　　下面是一个读取JSON文件的示例代码：
　　import json
　　with open('data.json', 'r', encoding='utf-8') as f:
　　 data = json.load(f)
　　 print(data)

　　在这个示例中，我们使用了json.load()函数来从文件中读取JSON数据。
　　如何自定义分隔符和编码
　　除了使用默认的分隔符和编码外，我们还可以根据需要自定义分隔符和编码。在csv模块中，通过设置dialect对象来实现自定义分隔符。例如，如果要使用制表符作为分隔符，可以使用如下代码：
　　import csv
　　csv.register_dialect('mydialect', delimiter='\t')
　　with open('data.csv', 'r', encoding='utf-8') as f:
　　 reader = csv.reader(f, dialect='mydialect')
　　 for row in reader:
　　 print(row)

　　在这个示例中，我们使用了csv.register_dialect()函数来注册一个新的dialect对象，并将分隔符设置为制表符。然后，在读取CSV文件时，通过设置dialect参数来指定使用该dialect对象。
　　在处理JSON文件时，如果需要自定义编码，则可以使用json.loads()函数来手动解码文件内容。例如，如果要将GBK编码的JSON文件转换为UTF-8编码的Python对象，可以使用如下代码：
　　import json
　　with open('data.json', 'r', encoding='gbk') as f:
　　 content = f.read()
　　 data = json.loads(content.encode('utf-8').decode('unicode_escape'))
　　 print(data)

　　在这个示例中，我们首先将文件内容读取出来，并使用encode()函数将其转换为bytes类型的数据。然后，使用decode()函数将其解码为unicode编码格式的字符串。最后，使用json.loads()函数将该字符串转换为Python对象。
　　pandas 中的 CSV 和 JSON 读写
　　pandas 模块的介绍和应用场景
　　pandas是Python中一个非常强大的数据处理库，它提供了灵活高效的数据结构和数据分析工具，适用于各种数据清洗、处理和分析的任务。其中，最常用的数据结构为DataFrame和Series，可以方便地处理各种表格型数据和时间序列数据。
　　pandas模块在读写CSV和JSON文件方面也提供了非常便捷的方法，比如read_csv()和read_json()函数可以自动将文件加载到DataFrame对象中，而to_csv()和to_json()函数则可以将DataFrame对象保存为CSV和JSON文件。
　　如何使用 pandas 读写 CSV 和 JSON 文件
　　在使用pandas读取CSV文件时，可以直接使用read_csv()函数加载文件并返回一个DataFrame对象。下面是一个读取CSV文件的示例代码：
　　import pandas as pd
　　df = pd.read_csv('data.csv', encoding='utf-8')
　　print(df.head())

　　在这个示例中，我们使用了pd.read_csv()函数来读取CSV文件，并将返回值赋给一个DataFrame对象。如果需要指定分隔符，则可以使用sep参数来设置。例如，如果分隔符为制表符，则可以使用如下代码：
　　import pandas as pd
　　df = pd.read_csv('data.csv', encoding='utf-8', sep='\t')
　　print(df.head())

　　在使用pandas读取JSON文件时，可以直接使用read_json()函数加载文件并返回一个DataFrame对象。下面是一个读取JSON文件的示例代码：
　　import pandas as pd
　　df = pd.read_json('data.json', encoding='utf-8')
　　print(df.head())

　　在这个示例中，我们使用了pd.read_json()函数来读取JSON文件，并将返回值赋给一个DataFrame对象。
　　在使用pandas保存DataFrame对象到CSV或JSON文件时，可以使用to_csv()和to_json()函数。例如，要将DataFrame对象保存为CSV文件，可以使用如下代码：
　　import pandas as pd
　　df = pd.DataFrame({'name': ['Alice', 'Bob'], 'age': [20, 30]})
　　df.to_csv('output.csv', index=False)

　　在这个示例中，我们先创建了一个简单的DataFrame对象，然后使用to_csv()函数将其保存为CSV文件。其中，index参数用于控制是否将行索引写入文件中（默认为True）。
　　要将DataFrame对象保存为JSON文件，可以使用类似的方式，例如：
　　import pandas as pd
　　df = pd.DataFrame({'name': ['Alice', 'Bob'], 'age': [20, 30]})
　　df.to_json('output.json', orient='records')

　　在这个示例中，我们使用了orient参数来指定JSON格式的输出方式。默认情况下，该参数为'columns'，表示以列为单位输出JSON数据。如果需要按行输出，则可以将其设置为'records'。
　　pandas 的数据清洗和转换功能
　　除了方便的读写CSV和JSON文件外，pandas还提供了大量的数据处理和转换功能，例如数据过滤、排序、分组、合并等等。下面是一个简单的示例代码，演示如何使用pandas对数据进行过滤和统计：
　　import pandas as pd
　　df = pd.read_csv('data.csv', encoding='utf-8')
　　df = df[df['age'] > 20]
　　print(df.groupby('gender')['height'].mean())

　　在这个示例中，我们首先使用pd.read_csv()函数加载CSV文件，并将返回值赋给一个DataFrame对象。然后，使用逻辑运算符筛选出年龄大于20岁的数据。最后，使用`groupby()`函数按照性别进行分组，并计算每个分组中身高的平均值。
　　除了以上示例，pandas还提供了非常丰富的数据清洗和转换功能，例如数据透视表、合并、重塑、填充缺失值等等。在实际应用中，可以根据具体需求选择相应的函数来完成数据处理任务。
　　总结
　　本篇博客介绍了Python中CSV和JSON格式的高级处理方法，包括如何识别和处理不同编码和分隔符的文件，以及如何使用pandas模块读写和处理CSV和JSON文件。在实际应用中，需要根据具体需求选择合适的函数和参数来完成数据处理任务。

欢迎光临 51Testing软件测试论坛 (http://bbs.51testing.com/)