Python读写txt文本文件

布满荆棘的人生 2023-10-18 17:11 147阅读 0赞

## [Python读写txt文本文件][Python_txt] ##

**一、文件的打开和创建**

<table> 
 <tbody> 
 <tr> 
 <td> 1 2 3 4 5 </td> 
 <td> <code>&gt;&gt;&gt; f </code><code>=</code>&nbsp;<code>open</code><code>(</code><code>'/tmp/test.txt'</code><code>)</code> <code>&gt;&gt;&gt; f.read()</code> <code>'hello python!\nhello world!\n'</code> <code>&gt;&gt;&gt; f</code> <code>&lt;</code><code>open</code>&nbsp;<code>file</code>&nbsp;<code>'/tmp/test.txt'</code><code>, mode </code><code>'r'</code>&nbsp;<code>at </code><code>0x7fb2255efc00</code><code>&gt;</code> </td> 
 </tr> 
 </tbody> 
</table>

**二、文件的读取**  
步骤：打开 -- 读取 -- 关闭

<table> 
 <tbody> 
 <tr> 
 <td> 1 2 3 4 </td> 
 <td> <code>&gt;&gt;&gt; f </code><code>=</code>&nbsp;<code>open</code><code>(</code><code>'/tmp/test.txt'</code><code>)</code> <code>&gt;&gt;&gt; f.read()</code> <code>'hello python!\nhello world!\n'</code> <code>&gt;&gt;&gt; f.close()</code> </td> 
 </tr> 
 </tbody> 
</table>

读取数据是后期数据处理的必要步骤。.txt是广泛使用的数据文件格式。一些.csv, .xlsx等文件可以转换为.txt 文件进行读取。我常使用的是Python自带的I/O接口，将数据读取进来存放在list中，然后再用numpy科学计算包将list的数据转换为array格式，从而可以像MATLAB一样进行科学计算。

下面是一段常用的读取txt文件代码，可以用在大多数的txt文件读取中

<table> 
 <tbody> 
 <tr> 
 <td> 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 </td> 
 <td> <code>filename </code><code>=</code>&nbsp;<code>'array_reflection_2D_TM_vertical_normE_center.txt'</code>&nbsp;<code># txt文件和当前脚本在同一目录下，所以不用写具体路径</code> <code>pos </code><code>=</code>&nbsp;<code>[]</code> <code>Efield </code><code>=</code>&nbsp;<code>[]</code> <code>with </code><code>open</code><code>(filename, </code><code>'r'</code><code>) as file_to_read:</code> <code>&nbsp;&nbsp;</code><code>while</code>&nbsp;<code>True</code><code>:</code> <code>&nbsp;&nbsp;&nbsp;&nbsp;</code><code>lines </code><code>=</code>&nbsp;<code>file_to_read.readline() </code><code># 整行读取数据</code> <code>&nbsp;&nbsp;&nbsp;&nbsp;</code><code>if</code>&nbsp;<code>not</code>&nbsp;<code>lines:</code> <code>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;</code><code>break</code> <code>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;</code><code>pass</code> <code>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;</code><code>p_tmp, E_tmp </code><code>=</code>&nbsp;<code>[</code><code>float</code><code>(i) </code><code>for</code>&nbsp;<code>i </code><code>in</code>&nbsp;<code>lines.split()] </code><code># 将整行数据分割处理，如果分割符是空格，括号里就不用传入参数，如果是逗号， 则传入‘，'字符。</code> <code>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;</code><code>pos.append(p_tmp)&nbsp; </code><code># 添加新读取的数据</code> <code>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;</code><code>Efield.append(E_tmp)</code> <code>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;</code><code>pass</code> <code>&nbsp;&nbsp;&nbsp;</code><code>pos </code><code>=</code>&nbsp;<code>np.array(pos) </code><code># 将数据从list类型转换为array类型。</code> <code>&nbsp;&nbsp;&nbsp;</code><code>Efield </code><code>=</code>&nbsp;<code>np.array(Efield)</code> <code>&nbsp;&nbsp;&nbsp;</code><code>pass</code> </td> 
 </tr> 
 </tbody> 
</table>

例如下面是将要读入的txt文件

![2016626171647895.png (429×301)][2016626171647895.png _429_301]

经过读取后，在Enthought Canopy的variable window查看读入的数据， 左侧为pos，右侧为Efield。  
  
![2016626171713978.png (148×277)][2016626171713978.png _148_277]![2016626171743777.png (147×280)][2016626171743777.png _147_280]

**三、文件写入（慎重，小心别清空原本的文件）**  
步骤：打开 -- 写入 -- （保存）关闭  
直接的写入数据是不行的，因为默认打开的是'r' 只读模式

<table> 
 <tbody> 
 <tr> 
 <td> 1 2 3 4 5 6 </td> 
 <td> <code>&gt;&gt;&gt; f.write(</code><code>'hello boy'</code><code>)</code> <code>Traceback (most recent call last):</code> <code>File</code>&nbsp;<code>"&lt;stdin&gt;"</code><code>, line </code><code>1</code><code>, </code><code>in</code>&nbsp;<code>&lt;module&gt;</code> <code>IOError: </code><code>File</code>&nbsp;<code>not</code>&nbsp;<code>open</code>&nbsp;<code>for</code>&nbsp;<code>writing</code> <code>&gt;&gt;&gt; f</code> <code>&lt;</code><code>open</code>&nbsp;<code>file</code>&nbsp;<code>'/tmp/test.txt'</code><code>, mode </code><code>'r'</code>&nbsp;<code>at </code><code>0x7fe550a49d20</code><code>&gt;</code> </td> 
 </tr> 
 </tbody> 
</table>

应该先指定可写的模式

<table> 
 <tbody> 
 <tr> 
 <td> 1 2 </td> 
 <td> <code>&gt;&gt;&gt; f1 </code><code>=</code>&nbsp;<code>open</code><code>(</code><code>'/tmp/test.txt'</code><code>,</code><code>'w'</code><code>)</code> <code>&gt;&gt;&gt; f1.write(</code><code>'hello boy!'</code><code>)</code> </td> 
 </tr> 
 </tbody> 
</table>

但此时数据只写到了缓存中，并未保存到文件，而且从下面的输出可以看到，原先里面的配置被清空了

关闭这个文件即可将缓存中的数据写入到文件中

<table> 
 <tbody> 
 <tr> 
 <td> 1 2 3 </td> 
 <td> <code>&gt;&gt;&gt; f1.close()</code> <code>[root@node1 ~]</code><code># cat /tmp/test.txt</code> <code>[root@node1 ~]</code><code># hello boy!</code> </td> 
 </tr> 
 </tbody> 
</table>

注意：这一步需要相当慎重，因为如果编辑的文件存在的话，这一步操作会先清空这个文件再重新写入。那么如果不要清空文件再写入该如何做呢？  
使用r+ 模式不会先清空，但是会替换掉原先的文件，如下面的例子：hello boy! 被替换成hello aay!

<table> 
 <tbody> 
 <tr> 
 <td> 1 2 3 4 5 </td> 
 <td> <code>&gt;&gt;&gt; f2 </code><code>=</code>&nbsp;<code>open</code><code>(</code><code>'/tmp/test.txt'</code><code>,</code><code>'r+'</code><code>)</code> <code>&gt;&gt;&gt; f2.write(</code><code>'\nhello aa!'</code><code>)</code> <code>&gt;&gt;&gt; f2.close()</code> <code>[root@node1 python]</code><code># cat /tmp/test.txt</code> <code>hello aay!</code> </td> 
 </tr> 
 </tbody> 
</table>

如何实现不替换？

<table> 
 <tbody> 
 <tr> 
 <td> 1 2 3 4 5 6 7 8 </td> 
 <td> <code>&gt;&gt;&gt; f2 </code><code>=</code>&nbsp;<code>open</code><code>(</code><code>'/tmp/test.txt'</code><code>,</code><code>'r+'</code><code>)</code> <code>&gt;&gt;&gt; f2.read()</code> <code>'hello girl!'</code> <code>&gt;&gt;&gt; f2.write(</code><code>'\nhello boy!'</code><code>)</code> <code>&gt;&gt;&gt; f2.close()</code> <code>[root@node1 python]</code><code># cat /tmp/test.txt</code> <code>hello girl!</code> <code>hello boy!</code> </td> 
 </tr> 
 </tbody> 
</table>

可以看到，如果在写之前先读取一下文件，再进行写入，则写入的数据会添加到文件末尾而不会替换掉原先的文件。这是因为指针引起的，r+ 模式的指针默认是在文件的开头，如果直接写入，则会覆盖源文件，通过read() 读取文件后，指针会移到文件的末尾，再写入数据就不会有问题了。这里也可以使用a 模式

<table> 
 <tbody> 
 <tr> 
 <td> 1 2 3 4 5 6 7 8 </td> 
 <td> <code>&gt;&gt;&gt; f </code><code>=</code>&nbsp;<code>open</code><code>(</code><code>'/tmp/test.txt'</code><code>,</code><code>'a'</code><code>)</code> <code>&gt;&gt;&gt; f.write(</code><code>'\nhello man!'</code><code>)</code> <code>&gt;&gt;&gt; f.close()</code> <code>&gt;&gt;&gt;</code> <code>[root@node1 python]</code><code># cat /tmp/test.txt</code> <code>hello girl!</code> <code>hello boy!</code> <code>hello man!</code> </td> 
 </tr> 
 </tbody> 
</table>

关于其他模式的介绍，见下表：

[![2016626170852899.png (713×317)][2016626170852899.png _713_317]][2016626170852899.png _713_317]

文件对象的方法：  
f.readline() 逐行读取数据  
方法一：

<table> 
 <tbody> 
 <tr> 
 <td> 1 2 3 4 5 6 7 8 9 </td> 
 <td> <code>&gt;&gt;&gt; f </code><code>=</code>&nbsp;<code>open</code><code>(</code><code>'/tmp/test.txt'</code><code>)</code> <code>&gt;&gt;&gt; f.readline()</code> <code>'hello girl!\n'</code> <code>&gt;&gt;&gt; f.readline()</code> <code>'hello boy!\n'</code> <code>&gt;&gt;&gt; f.readline()</code> <code>'hello man!'</code> <code>&gt;&gt;&gt; f.readline()</code> <code>''</code> </td> 
 </tr> 
 </tbody> 
</table>

方法二：

<table> 
 <tbody> 
 <tr> 
 <td> 1 2 3 4 5 6 7 8 9 10 11 12 </td> 
 <td> <code>&gt;&gt;&gt; </code><code>for</code>&nbsp;<code>i </code><code>in</code>&nbsp;<code>open</code><code>(</code><code>'/tmp/test.txt'</code><code>):</code> <code>...&nbsp;&nbsp; </code><code>print</code>&nbsp;<code>i</code> <code>...</code> <code>hello girl!</code> <code>hello boy!</code> <code>hello man!</code> <code>f.readlines()&nbsp;&nbsp; 将文件内容以列表的形式存放</code> &nbsp; <code>&gt;&gt;&gt; f </code><code>=</code>&nbsp;<code>open</code><code>(</code><code>'/tmp/test.txt'</code><code>)</code> <code>&gt;&gt;&gt; f.readlines()</code> <code>[</code><code>'hello girl!\n'</code><code>, </code><code>'hello boy!\n'</code><code>, </code><code>'hello man!'</code><code>]</code> <code>&gt;&gt;&gt; f.close()</code> </td> 
 </tr> 
 </tbody> 
</table>

f.next() 逐行读取数据，和f.readline() 相似，唯一不同的是，f.readline() 读取到最后如果没有数据会返回空，而f.next() 没读取到数据则会报错

f.writelines() 多行写入

<table> 
 <tbody> 
 <tr> 
 <td> 1 2 3 4 5 6 7 8 9 10 11 </td> 
 <td> <code>&gt;&gt;&gt; l </code><code>=</code>&nbsp;<code>[</code><code>'\nhello dear!'</code><code>,</code><code>'\nhello son!'</code><code>,</code><code>'\nhello baby!\n'</code><code>]</code> <code>&gt;&gt;&gt; f </code><code>=</code>&nbsp;<code>open</code><code>(</code><code>'/tmp/test.txt'</code><code>,</code><code>'a'</code><code>)</code> <code>&gt;&gt;&gt; f.writelines(l)</code> <code>&gt;&gt;&gt; f.close()</code> <code>[root@node1 python]</code><code># cat /tmp/test.txt</code> <code>hello girl!</code> <code>hello boy!</code> <code>hello man!</code> <code>hello dear!</code> <code>hello son!</code> <code>hello baby!</code> </td> 
 </tr> 
 </tbody> 
</table>

f.seek(偏移量，选项)

这个例子可以充分的解释前面使用r+这个模式的时候，为什么需要执行f.read()之后才能正常插入  
f.seek(偏移量，选项)  
（1）选项=0，表示将文件指针指向从文件头部到“偏移量”字节处  
（2）选项=1，表示将文件指针指向从文件的当前位置，向后移动“偏移量”字节  
（3）选项=2，表示将文件指针指向从文件的尾部，向前移动“偏移量”字节

偏移量：正数表示向右偏移，负数表示向左偏移

<table> 
 <tbody> 
 <tr> 
 <td> 1 2 3 4 5 6 7 8 9 10 11 12 13 </td> 
 <td> <code>&gt;&gt;&gt; f </code><code>=</code>&nbsp;<code>open</code><code>(</code><code>'/tmp/test.txt'</code><code>,</code><code>'r+'</code><code>)</code> <code>&gt;&gt;&gt; f.seek(</code><code>0</code><code>,</code><code>2</code><code>)</code> <code>&gt;&gt;&gt; f.readline()</code> <code>''</code> <code>&gt;&gt;&gt; f.seek(</code><code>0</code><code>,</code><code>0</code><code>)</code> <code>&gt;&gt;&gt; f.readline()</code> <code>'hello girl!\n'</code> <code>&gt;&gt;&gt; f.readline()</code> <code>'hello boy!\n'</code> <code>&gt;&gt;&gt; f.readline()</code> <code>'hello man!\n'</code> <code>&gt;&gt;&gt; f.readline()</code> <code>''</code> </td> 
 </tr> 
 </tbody> 
</table>

f.flush() 将修改写入到文件中（无需关闭文件）

<table> 
 <tbody> 
 <tr> 
 <td> 1 2 </td> 
 <td> <code>&gt;&gt;&gt; f.write(</code><code>'hello python!'</code><code>)</code> <code>&gt;&gt;&gt; f.flush()</code> </td> 
 </tr> 
 </tbody> 
</table>

<table> 
 <tbody> 
 <tr> 
 <td> 1 </td> 
 <td> <code>[root@node1 python]</code><code># cat /tmp/test.txt</code> </td> 
 </tr> 
 </tbody> 
</table>

<table> 
 <tbody> 
 <tr> 
 <td> 1 2 3 4 </td> 
 <td> <code>hello girl!</code> <code>hello boy!</code> <code>hello man!</code> <code>hello python!</code> </td> 
 </tr> 
 </tbody> 
</table>

f.tell() 获取指针位置

<table> 
 <tbody> 
 <tr> 
 <td> 1 2 3 4 5 6 7 8 9 </td> 
 <td> <code>&gt;&gt;&gt; f </code><code>=</code>&nbsp;<code>open</code><code>(</code><code>'/tmp/test.txt'</code><code>)</code> <code>&gt;&gt;&gt; f.readline()</code> <code>'hello girl!\n'</code> <code>&gt;&gt;&gt; f.tell()</code> <code>12</code> <code>&gt;&gt;&gt; f.readline()</code> <code>'hello boy!\n'</code> <code>&gt;&gt;&gt; f.tell()</code> <code>23</code> </td> 
 </tr> 
 </tbody> 
</table>

**四、内容查找和替换  
1、内容查找**  
实例：统计文件中hello个数  
思路：打开文件，遍历文件内容，通过正则表达式匹配关键字，统计匹配个数。

脚本如下：  
方法一：

<table> 
 <tbody> 
 <tr> 
 <td> 1 2 3 4 5 6 7 8 9 10 </td> 
 <td> <code>#!/usr/bin/python</code> <code>import</code>&nbsp;<code>re</code> <code>f </code><code>=</code>&nbsp;<code>open</code><code>(</code><code>'/tmp/test.txt'</code><code>)</code> <code>source </code><code>=</code>&nbsp;<code>f.read()</code> <code>f.close()</code> <code>r </code><code>=</code>&nbsp;<code>r</code><code>'hello'</code> <code>s </code><code>=</code>&nbsp;<code>len</code><code>(re.findall(r,source))</code> <code>print</code>&nbsp;<code>s</code> <code>[root@node1 python]</code><code># python count.py</code> <code>4</code> </td> 
 </tr> 
 </tbody> 
</table>

方法二：

<table> 
 <tbody> 
 <tr> 
 <td> 1 2 3 4 5 6 7 8 9 10 11 12 </td> 
 <td> <code>#!/usr/bin/python</code> <code>import</code>&nbsp;<code>re</code> <code>fp </code><code>=</code>&nbsp;<code>file</code><code>(</code><code>"/tmp/test.txt"</code><code>,</code><code>'r'</code><code>)</code> <code>count </code><code>=</code>&nbsp;<code>0</code> <code>for</code>&nbsp;<code>s </code><code>in</code>&nbsp;<code>fp.readlines():</code> <code>li </code><code>=</code>&nbsp;<code>re.findall(</code><code>"hello"</code><code>,s)</code> <code>if</code>&nbsp;<code>len</code><code>(li)&gt;</code><code>0</code><code>:</code> <code>count </code><code>=</code>&nbsp;<code>count </code><code>+</code>&nbsp;<code>len</code><code>(li)</code> <code>print</code>&nbsp;<code>"Search"</code><code>,count, </code><code>"hello"</code> <code>fp.close()</code> <code>[root@node1 python]</code><code># python count1.py</code> <code>Search </code><code>4</code>&nbsp;<code>hello</code> </td> 
 </tr> 
 </tbody> 
</table>

**2、替换**  
实例：把test.txt 中的hello全部换为"hi"，并把结果保存到myhello.txt中。

<table> 
 <tbody> 
 <tr> 
 <td> 1 2 3 4 5 6 7 8 9 10 11 12 13 14 </td> 
 <td> <code>#!/usr/bin/python</code> <code>import</code>&nbsp;<code>re</code> <code>f1 </code><code>=</code>&nbsp;<code>open</code><code>(</code><code>'/tmp/test.txt'</code><code>)</code> <code>f2 </code><code>=</code>&nbsp;<code>open</code><code>(</code><code>'/tmp/myhello.txt'</code><code>,</code><code>'r+'</code><code>)</code> <code>for</code>&nbsp;<code>s </code><code>in</code>&nbsp;<code>f1.readlines():</code> <code>f2.write(s.replace(</code><code>'hello'</code><code>,</code><code>'hi'</code><code>))</code> <code>f1.close()</code> <code>f2.close()</code> <code>[root@node1 python]</code><code># touch /tmp/myhello.txt</code> <code>[root@node1 ~]</code><code># cat /tmp/myhello.txt</code> <code>hi girl!</code> <code>hi boy!</code> <code>hi man!</code> <code>hi python!</code> </td> 
 </tr> 
 </tbody> 
</table>

实例：读取文件test.txt内容，去除空行和注释行后，以行为单位进行排序，并将结果输出为result.txt。test.txt 的内容如下所示：

<table> 
 <tbody> 
 <tr> 
 <td> 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 </td> 
 <td> <code>#some words</code> &nbsp; <code>Sometimes in life,</code> <code>You find a special friend;</code> <code>Someone who changes your life just by being part of it.</code> <code>Someone who makes you laugh until you can't stop;</code> <code>Someone who makes you believe that there really is good in the world.</code> <code>Someone who convinces you that there really is an unlocked door just waiting for you to open it.</code> <code>This is Forever Friendship.</code> <code>when you're down,</code> <code>and the world seems dark and empty,</code> <code>Your forever friend lifts you up in spirits and makes that dark and empty world</code> <code>suddenly seem bright and full.</code> <code>Your forever friend gets you through the hard times,the sad times,and the confused times.</code> <code>If you turn and walk away,</code> <code>Your forever friend follows,</code> <code>If you lose you way,</code> <code>Your forever friend guides you and cheers you on.</code> <code>Your forever friend holds your hand and tells you that everything is going to be okay. </code> </td> 
 </tr> 
 </tbody> 
</table>

脚本如下：

<table> 
 <tbody> 
 <tr> 
 <td> 1 2 3 4 5 6 7 8 9 10 </td> 
 <td> <code>f </code><code>=</code>&nbsp;<code>open</code><code>(</code><code>'cdays-4-test.txt'</code><code>)</code> <code>result </code><code>=</code>&nbsp;<code>list</code><code>()</code> <code>for</code>&nbsp;<code>line </code><code>in</code>&nbsp;<code>f.readlines():&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp; </code><code># 逐行读取数据</code> <code>line </code><code>=</code>&nbsp;<code>line.strip()&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp; </code><code>#去掉每行头尾空白</code> <code>if</code>&nbsp;<code>not</code>&nbsp;<code>len</code><code>(line) </code><code>or</code>&nbsp;<code>line.startswith(</code><code>'#'</code><code>):&nbsp;&nbsp; # 判断是否是空行或注释行</code> <code>continue</code>&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;<code>#是的话，跳过不处理</code> <code>result.append(line)&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp; </code><code>#保存</code> <code>result.sort()&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp; </code><code>#排序结果</code> <code>print</code>&nbsp;<code>result</code> <code>open</code><code>(</code><code>'cdays-4-result.txt'</code><code>,</code><code>'w'</code><code>).write(</code><code>'%s'</code>&nbsp;<code>%</code>&nbsp;<code>'\n'</code><code>.join(result))&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp;&nbsp; </code><code>#保存入结果文件</code> </td> 
 </tr> 
 </tbody> 
</table>

原文链接: [https://www.cnblogs.com/hackpig/p/8215786.html][Python_txt] 侵删

[Python_txt]: https://www.cnblogs.com/hackpig/p/8215786.html
[2016626171647895.png _429_301]: http://files.jb51.net/file_images/article/201606/2016626171647895.png?201652617172
[2016626171713978.png _148_277]: http://files.jb51.net/file_images/article/201606/2016626171713978.png?2016526171725
[2016626171743777.png _147_280]: http://files.jb51.net/file_images/article/201606/2016626171743777.png?2016526171754
[2016626170852899.png _713_317]: http://files.jb51.net/file_images/article/201606/2016626170852899.png?20165261792