Python的txt、csv、np、json格式文件读取和写入

Love The Way You Lie 2022-10-11 15:47 228阅读 0赞

**文件的读写模式**

*  r : 读取文件，若文件不存在则会报错
 *  w: 写入文件，若文件不存在则会先创建再写入，会覆盖原文件
 *  a : 写入文件，若文件不存在则会先创建再写入，但不会覆盖原文件，而是追加在文件末尾
 *  rb,wb：分别与r，w类似，但是用于读写二进制文件
 *  r+ : 可读、可写，文件不存在也会报错，写操作时会覆盖
 *  w+ : 可读，可写，文件不存在先创建，会覆盖
 *  a+ : 可读、可写，文件不存在先创建，不会覆盖，追加在末尾

## .txt ##

数据为字符串格式

# 写入
    with open(filename, "w", encoding='utf-8') as f:
        f.write("打瞌睡")  
    
    # 读取
    with open(filename, 'r', encoding='utf-8') as f:
        data1 = f.read()  # 读取文本的全部内容，以字符串形式返回
        print('data1:', data1)
    with open(filename, 'r', encoding='utf-8') as f:
        data2 = f.readline()  # 读取文本的第一行内容，以字符串形式返回
        print('data2:', data2)
    with open(filename, 'r', encoding='utf-8') as f:
        data3 = f.readlines()  # 读取文本所有内容，以列表形式返回
        print('data3:', data3)
        for line in data3:
            line = line.strip('\n')
            print(line)
    """ data1: da ke shui 打瞌睡 data2: da data3: ['da\n', 'ke\n', 'shui\n', '打瞌睡'] da ke shui 打瞌睡 """

## .csv ##

数据为字符串格式

import pandas as pd
    import csv
    
    # 写入，newline=''解决空行问题
    with open(filename, "w", encoding='utf-8', newline='') as csvfile:
        writer = csv.writer(csvfile)
        writer.writerow(["姓名", "班级", "学号"])
        
    # 读取，第一行为抬头信息，不作为数据读取
    df = pd.read_csv(filename, header=0, encoding='utf-8', engine='python')
    for i in range(len(df)):
        name = df["姓名"][i]
        grade = df["班级"][i]
        id = df["学号"][i]

## .np ##

数据为ndarray格式，可保留内部数据格式  
文件不能追加写入，只能覆盖写入  
npy为二进制文件，npz为压缩文件

import numpy as np
    
    # 写入
    np.save(filename, np.array([[1, 2, 3], [4, 5, 6]]))     # 将数组以.npy形式存储在二进制文件中
    np.savez(filename, np.array([1, 2]), np.array([3, 4]))    # 以.npz格式将多个数组保存到单个文件中
    
    # 读取
    data = np.load(filename, encoding="latin1", allow_pickle=True)
    txt_data = np.loadtxt(filename, delimiter=',', unpack=True)
    """ delimiter分隔符 unpack用多个变量接受数据 """

## .json ##

数据为json格式，可保留内部数据格式

import json
    
    # 写入
    with open(filename, "w", encoding='utf-8') as f:
        f.write(json.dumps({ 'id': [1, 2]}, ensure_ascii=False))
    
    # 读取
    with open(filename, 'r', encoding='utf-8') as f:
        data = json.loads(f)

注：

*  `dump()`和`dumps()`对python对象进行序列化，将python对象进行json格式的编码。
 *  `load()`和`loads()`为反序列化方法，将json格式数据解码为python对象。
 *  `dumps()`和loads()函数不需要传文件描述符，其他参数和`dump()`、`load()`函数一样。
 *  文件描述符 fp，json模块总是生成str对象，而不是字节对象，因此`fp.write()`必须支持str输入。