python之txt文件基本操作

1. 读取txt文本内容

python常用的读取文件函数有三种read()、readline()、readlines()

1.1 read()

read()一次性读取文本中全部的内容，以字符串的形式返回结果.

with open('test.txt', 'r') as f:  # 打开文件
    data = f.read()  # 读取文件
    print(data)

1.2 readline()

readline() 只读取文本第一行的内容，以字符串的形式返回结果.

with open('test.txt', 'r') as f:
    data = f.readline()
    print(data)

1.3 readlines()

readlines() 读取文本所有内容，并且以列表的格式返回结果，一般配合for in使用.

with open('test.txt', 'r') as f:
    data = f.readlines()
    print(data)

readlines会读到换行符，可用如下方法去除：

with open('test.txt', 'r') as f:
    for line in f.readlines():
        line = line.strip('\n')  #去掉列表中每一个元素的换行符
        print(line)

2. 写入txt文本

with open('test.txt','w') as f:
    f.write("写入文件测试！")  # 自带文件关闭功能，不需要再写f.close()

2.1 读写模式

要了解文件读写模式，需要了解几种模式的区别，以及对应指针

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

2.2 读取txt文件的若干行写入另一个txt

方法一：

file_1 = './data/1.txt'
file_2 = './data/2.txt'

with open(file_1, 'rb') as fp1:
    with open(file_2, 'ab') as fp2:
        for i in range(100):
            line_data = fp1.readline()
            fp2.write(line_data)

方法二：

file_1 = './data/1.txt'
file_2 = './data/2.txt'

fp_1 = open(file_1, 'rb')
fp_2 = open(file_2, 'ab')
 
line_id = 0
while True:
    line_data = fp_1.readline()
    # print('line_data:', line_data)
    line_id = line_id + 1
    
    if line_id>=1 and line_id<=100:
        print('line_id:', line_id)
        fp_2.write(line_data)
    else:
        break
        
fp_1.close() # 文件关闭, 若不关闭可能会导致部分写入数据丢失
fp_2.close() # 文件关闭, 若不关闭可能会导致部分写入数据丢失

3. 其他使用

3.1 删掉指定字符前/后的内容

如下所示, 我们想删除右括号 ] 左边的内容, 得到 Today is Wednesday

line_data = '2021-09-01 [] Today is Wednesday'
head, sep, tail = line_data.partition('] ')
print(tail)

输出结果如下所示:

Today is Wednesday

3.2 csv文件存为txt

使用df.to_csv()不仅可以存储CSV文件, 还可以存储TXT文件:

import pandas as pd

df = pd.read_csv('./data/test.csv', header=None)
df.to_csv('./data/test.txt', sep=' ', header=None, index=None) # sep=' ' 分隔符为空格

3.3 numpy读写txt

3.3.1 np.savetxt()

numpy.savetxt(fname, X, fmt='%.18e', delimiter=' ', newline='\n', header='', footer='', comments='# ', encoding=None)[source]

• fname：表示要保存文件的地址，可以自己建文件名，如‘test.txt’
• X：表示要保存的文件
• fmt:控制数据存储的格式
• delimiter ：分隔符，默认空格，也可以用逗号等
• newline:表示换行的时候用什么，默认\n，表示换一行，也可以用\t，则表示空四格
• header：表示头文件，如“test_data"
• footer: 文件下的脚注
• comment：注释，默认是#,因为python的注释是#，也可以用其它符号

样例代码如下：

import numpy as np 
#生成数据 
x = y = z = np.ones((2,3)) 
x 
>>> array([[1., 1., 1.], 
       [1., 1., 1.]]) 
        
#保存数据 
np.savetxt('test.out', x) 
np.savetxt('test1.out', x, fmt='%d') 
np.savetxt('test2.out', x, delimiter=',') 
np.savetxt('test3.out', x, newline='a') 
np.savetxt('test4.out', x, delimiter=',',newline='a') 
np.savetxt('test5.out', x, delimiter=',',header='abc') 
np.savetxt('test6.out', x, delimiter=',',footer='abc') 

3.3.2 np.loadtxt()

numpy.loadtxt(fname, dtype=<class 'float'>, comments='#', delimiter=None, converters=None, skiprows=0, usecols=None, unpack=False, ndmin=0, encoding='bytes')

• fname:文件名/文件路径,如果文件后缀是.gz或.bz2,文件将被解压,然后再载入
• dtype:要读取的数据类型
• comments:文件头部或者尾部字符串的开头字符,用于识别头部,尾部字符串
• delimiter:划分读取上来值的字符串
• converters:数据行之间的分隔符

样例代码如下：

np.loadtxt('test.out') 
np.loadtxt('test2.out', delimiter=',')