python中实现txt文件读写

本文介绍了一种Python函数,用于从指定文件夹中读取所有TXT文件的数据。该函数允许从特定行开始读取,并提供了两种数据存储模式:一种不考虑行列顺序,另一种则按照TXT文件中的行列顺序存储数据。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

def read_txt(path,pass_n,model=0):
    import os
    import numpy as np
    #function:读取文件夹内txt文件
    #path:str,txt文件所在文件夹
    #pass_n:int,从txt文件中第pass_n行开始读文件
    #model:int,两个模式,model=0,不按行列顺序存储数据,model=1,按行列顺序存储数据
    
    #datas:array,输出txt内数据到数组
    datas=[]
    #分隔符
    sym1="\n" 
    sym2=","      #此处为txt文件中每个数据的分隔符,可根据需要自己修改
    x_dirs=os.listdir(path)
    for x_file in x_dirs:
        f_path=os.path.join(path,x_file)
        x_text=open(f_path)
        for i in range(pass_n):
            next(x_text)
        
        if model == 0:
            #按txt文件中数据不分行列存储,形状为(m,n)
            #m为文件数,n为每个txt内的数据量
            x_data=x_text.read()
            x_text.close()
            #去除末尾空格与‘\n’
            x_data=x_data.strip()
            x_data=x_data.replace(sym1,sym2)
            #按分隔符划分数据
            x_data=x_data.split(sym2)
            x_data=np.array(x_data)
        elif model == 1:
        
            #txt文件中数据按行存储,形状为(m,n_r,n_c)
            #m为txt文件个数。每个txt文件中有n_r行,n_c列数据
            x_data=x_text.readlines()
            x_text.close()
            #对同一行数据进行处理
            for i in range(len(x_data)):
                #print(type(x_data[i]))
                x_data[i]=x_data[i].strip()
                x_data[i]=x_data[i].replace(sym1,sym2)
                #按分隔符划分数据
                x_data[i]=x_data[i].split(sym2)
        else:
            print("请输入正确的模式:")
            print("model=0:按文件排列")
            print("model=1:按行列排列")
            
        datas.append(x_data)
    #将datas转化为数组
    datas=np.array(datas)
    print(datas.shape)
    
    return datas
                
Python实现txt文件读写可以使用内置的`open`函数配合文件操作模式来完成。以下是一个基本的介绍: ### 写入文件: 要写入文件,你可以使用`open`函数并指定文件路径和模式。模式“w”表示写入模式,如果文件不存在将会被创建,如果文件已存在,其内容将被覆盖。 ```python # 写入文件的代码示例 file_path = 'example.txt' # 指定文件路径 # 使用写入模式打开文件 with open(file_path, 'w', encoding='utf-8') as file: file.write("这是第一行文本。\n") file.write("这是第二行文本。\n") file.write("这是第三行文本。") ``` 在这个示例中,`with`语句用于确保文件正确关闭,即使在写入时发生异常也能保证文件资源的释放。`encoding='utf-8'`用于确保正确处理Unicode字符。 ### 读取文件: 要读取文件,可以使用相同的`open`函数,但是模式需要改为“r”表示只读模式。如果你想要读取整个文件内容为一个字符串,可以使用`read`方法。 ```python # 读取文件的代码示例 file_path = 'example.txt' # 指定文件路径 # 使用读取模式打开文件 with open(file_path, 'r', encoding='utf-8') as file: content = file.read() print(content) ``` 如果文件很大,一次性读取整个文件可能会消耗大量内存,这时可以逐行读取。 ```python with open(file_path, 'r', encoding='utf-8') as file: for line in file: print(line.strip()) # 使用strip()去除每行末尾的换行符 ``` ### 小结: 使用`open`函数配合不同的模式,可以轻松实现Python中的文件读写操作。记住使用`with`语句来管理文件资源,以避免文件未被正确关闭的问题。同时,正确设置编码方式(`encoding`参数),以确保文本的正确读写
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值