Python将字典写入txt文件|极客笔记

Python将字典写入txt文件

在Python编程中，我们有时会需要将数据输出到文本文件中以备后续使用。而在输出数据时，字典是一种常见的数据结构类型。Python提供了一个简单的方法来将字典写入文本文件中，本文将为您详细介绍如何实现。

准备工作

在介绍具体使用方法之前，我们需要先了解两个Python标准库：json和pickle。json库可以将Python对象序列化为字符串，并输出为文本文件。pickle库则可以将Python对象序列化为二进制文件。在本文中，我们将使用json库来将字典写入txt文件中。

代码示例

下面是一个简单的代码示例，演示了如何将字典写入txt文件中：

import json

data = {'name': '小明', 'age': 18, 'gender': '男'}
with open('data.txt', 'w') as f:
    json.dump(data, f)

在以上代码中，我们首先创建了一个字典对象data，该字典包含了一个人的姓名、年龄和性别信息。接着，我们使用Python的with语句以写入模式（’w’）打开了一个名为data.txt的文本文件，并将data对象写入该文件中。

在写入过程中，我们使用了json库的dump()函数，这个函数将字典data转换为一个字符串格式，并将其写入到txt文件中。因为json库操作的是字符串，所以我们可以将这个字符串当做普通文本一样地输出到文件中。

在文件输出完成后，我们可以打开这个文本文件来查看其内容：

{"name": "\u5c0f\u660e", "age": 18, "gender": "\u7537"}

可以看到，被写入的字典数据被转化为了一个JSON格式的字符串，并以Unicode编码输出。

容错处理

在进行字典写入操作时，有时候需要注意容错处理问题。比如，如果输入的字典中包含了不能被JSON编码的数据类型，比如set和tuple等类型，就会导致写入操作出错。

以下示例将在代码中添加一个set对象，用于演示此类问题：

import json

data = {'name': '小明', 'age': 18, 'gender': '男', 'hobby': {'music', 'reading'}}
with open('data.txt', 'w') as f:
    json.dump(data, f)

在以上代码中，我们向字典data中添加了一个名为hobby的set对象，用于表示小明的爱好。由于set对象不能被JSON编码，所以在写入文件时，程序会报错：

TypeError: Object of type 'set' is not JSON serializable

为了避免这种情况，我们可以在写入文件之前，先将字典中的非JSON编码类型进行转换。具体方法就是在字典转换到字符串的过程中，添加default参数，递归地将非JSON类型数据转换为JSON编码类型的数据。以下是一个示例代码：

import json

data = {'name': '小明', 'age': 18, 'gender': '男', 'hobby': {'music', 'reading'}}

def convert_set(data):
    if isinstance(data, set):
        return list(data)
    elif isinstance(data, dict):
        return dict(map(convert_set, data.items()))
    elif isinstance(data, (list, tuple)):
        return type(data)(map(convert_set, data))
    else:
        return data

with open('data.txt', 'w') as f:
    json.dump(convert_set(data), f)

在以上代码中，我们添加了一个convert_set函数，用于将字典中的“set对象转换为可以被JSON编码的list类型。在函数中，我们首先使用isinstance()函数判断数据类型，如果是set类型，就将其转换为list类型；如果是字典类型，就递归地处理其items()，并返回转换后的字典；如果是list或tuple`类型，就递归地处理其中的元素；否则，就返回原数据。

在调用json.dump()函数写入文件时，我们调用了convert_set()函数，让其返回值成为被写入的字符串：

with open('data.txt', 'w') as f:
    json.dump(convert_set(data), f)

这样，在写入文件时，convert_set()函数会被递归地执行，将字典中的set对象转换为list类型，保证了程序的正确性。

结论

在Python编程中，我们通常会需要将数据输出到文本文件中以备后续使用。而对于字典类型的数据，可以使用json库将其转换为JSON字符串，并输出到txt文件中。但需要注意的是，如果字典中包含了不能被JSON编码的数据类型，比如set和tuple等类型，就需要进行相应的容错处理，以确保程序的正确性。