Python 存储数据

很多程序都要求用户输入某种信息,如让用户存储游戏首选项或提供要可视化的数据。不管专注的是什么,程序都把用户提供的信息存储在列表和字典等数据结构中。用户关闭程序时,你几乎总是要保存他们提供的信息;一种简单的方式是使用模块json来存储数据。

模块json让你能够将简单的Python数据结构转储到文件中,并在程序再次运行时加载该文件中的数据。你还可以使用json在Python程序之间分享数据。更重要的是,JSON数据格式并非Python专用的,这让你能够将以JSON格式存储的数据与使用其他编程语言的人分享。这是一种轻便格式,很有用,也易于学习。

注意 JSON(JavaScript Object Notation)格式最初是为JavaScript开发的,但随后成了一种常见格式,被包括Python在内的众多语言采用。

json.dump()和json.load()

我们来编写一个存储一组数字的简短程序,再编写一个将这些数字读取到内存中的程序。第一个程序将使用json.dump()来存储这组数字,而第二个程序将使用json.load()。

函数json.dump()接受两个实参:要存储的数据以及可用于存储数据的文件对象。下面演示了如何使用json.dump()来存储数字列表:

import json
numbers = [2, 3, 5, 7, 11, 13]
filename = 'coolcou.json' #(1)
with open(filename, 'w') as file_object: #(2)
    json.dump(numbers , file_object) #(3)

我们先导入模块json,再创建一个数字列表。在(1)处,我们指定了要将该数字列表存储到其中的文件的名称。通常使用文件扩展名.json来指出文件存储的数据为JSON格式。接下来,我们以写入模式打开这个文件,让json能够将数据写入其中(见(2))。在(3)处,我们使用函数json.dump()将数字列表存储到文件coolcou.json中。

这个程序没有输出,但我们可以打开文件coolcou.json,看看其内容。数据的存储格式与Python中一样:

[2, 3, 5, 7, 11, 13]

下面再编写一个程序,使用json.load()将这个列表读取到内存中:

import json
filename = 'coolcou.json' #(1)
with open(filename) as file_object: #(2)
    numbers = json.load(file_object) #(3)
print(numbers)

(1)处,我们确保读取的是前面写入的文件。这次我们以读取方式打开这个文件,因为Python只需读取这个文件(见(2))。在(3)处,我们使用函数json.load()加载存储在coolcou.json中的信息,并将其存储到变量numbers中。最后,我们打印恢复的数字列表:

[2, 3, 5, 7, 11, 13]

这是一种在程序之间共享数据的简单方式。

保存和读取用户生成的数据

对于用户生成的数据,使用json保存它们大有裨益,因为如果不以某种方式进行存储,等程序停止运行时用户的信息将丢失。下面来看一个这样的例子:用户首次运行程序时被提示输入自己的名字,这样再次运行程序时就记住他了。

我们先来存储用户的名字:

import json
username = input("What is your name?")
filename = 'coolcou.json'
with open(filename, 'w') as file_object:
    json.dump(username, file_object)
    print("We'll remember you when you come back, " + username + "!")

输出如下:
Python 存储数据

现在再编写一个程序,向其名字被存储的用户发出问候:

import json
filename = 'coolcou.json'
with open(filename) as file_object:
    username = json.load(file_object) #(1)
    print("Welcome back, "+ username) #(2)

(1)处,我们使用json.load()将存储在coolcou.json中的信息读取到变量username中。恢复用户名后,我们就可以欢迎用户回来了(见(2)):

Welcome back, www.coolcou.com
赞(0)

评论 抢沙发

评论前必须登录!