51Testing软件测试论坛

 找回密码
 (注-册)加入51Testing

QQ登录

只需一步,快速开始

微信登录,快人一步

手机号码,快捷登录

查看: 674|回复: 0
打印 上一主题 下一主题

[python] Python CSV、 JSON 格式高级处理(下)

[复制链接]
  • TA的每日心情
    无聊
    3 天前
  • 签到天数: 1050 天

    连续签到: 1 天

    [LV.10]测试总司令

    跳转到指定楼层
    1#
    发表于 2023-6-9 11:24:00 | 只看该作者 回帖奖励 |倒序浏览 |阅读模式
    分隔符和编码
      了解不同分隔符和编码的应用
      接上篇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文件。在实际应用中,需要根据具体需求选择合适的函数和参数来完成数据处理任务。

    分享到:  QQ好友和群QQ好友和群 QQ空间QQ空间 腾讯微博腾讯微博 腾讯朋友腾讯朋友
    收藏收藏
    回复

    使用道具 举报

    本版积分规则

    关闭

    站长推荐上一条 /1 下一条

    小黑屋|手机版|Archiver|51Testing软件测试网 ( 沪ICP备05003035号 关于我们

    GMT+8, 2024-11-24 18:19 , Processed in 0.069014 second(s), 24 queries .

    Powered by Discuz! X3.2

    © 2001-2024 Comsenz Inc.

    快速回复 返回顶部 返回列表