Python如何用json模块存储数据

存储数据

很多程序都要求用户输入某种信息，程序把用户提供的信息存储在列表和字典等数据结构中。用户关闭程序时，就要保存提供的信息，一种简单的方式就是使用模块json来存储数据。

模块json能将简单的python数据结构存储到文件中，并在程序再次运转时加载该文件中的数据。还可以使用json在python程序之间分享数据，与使用其他编程语言的人分享。

1. 使用json.dump( )和json.load( )

importjson
numbers=[2,3,5,7,11,13]
filename='number.json'
withopen(filename,'w')asf_ojb:#以写入模式打开文件
json.dump(numbers,f_ojb)#使用函数json.dump()将列表存储到文件中
withopen(filename)asf_ojb:
nums=json.load(f_ojb)#使用函数json.load()将这个列表读取到内存中
print(nums)#打印读取到内存中的列表，比较是否与存入的列表相同

运行结果：

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

相关推荐：《Python视频教程》

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

importjson
#存储用户的名字
username=input('Whatisyourname?')
filename='username.json'
withopen(filename,'w')asf_obj:
json.dump(username,f_obj)#存储用户名与username.json文件中
print("We'llrememberyouwhenyoucomeback,"+username+"!")
#向名字被存储的用户发出问候
withopen(filename)asf_obj:
un=json.load(f_obj)
print("\nWelcomeback,"+un+"!")

运行结果：

Whatisyourname?ela
We'llrememberyouwhenyoucomeback,ela!
Welcomeback,ela!

优化上述代码：

importjson
#存储用户的名字
username=input('Whatisyourname?')
filename='username.json'
withopen(filename,'w')asf_obj:
json.dump(username,f_obj)#存储用户名与username.json文件中
print("We'llrememberyouwhenyoucomeback,"+username+"!")
#向名字被存储的用户发出问候
withopen(filename)asf_obj:
un=json.load(f_obj)
print("\nWelcomeback,"+un+"!")

运行结果：

Whatisyourname?ela
We'llrememberyouwhenyoucomeback,ela!
Welcomeback,ela!

优化上述代码：

importjson
#若存储了用户名就加载；否则提示用户输入并存储
filename='username.json'
try:
withopen(filename)asf_obj:
username=json.load(f_obj)
exceptFileNotFoundError:
username=input('Whatisyourname?')
withopen(filename,'w')asf_obj:
json.dump(username,f_obj)
print("We'llrememberyouwhenyoucomeback,"+username+"!")
else:
print("\nWelcomeback,"+username+"!")

运行结果：

Welcomeback,ela!

3. 重构

代码可以运行，但也可以做进一步改进——将代码划分成一些列完成具体工作的函数：这个过程称为重构。

目的：让代码更清晰、易于理解、易扩展。

importjson
defget_stored_username():
"""如果存储了用户名，就获取它"""
filename='username.json'
try:
withopen(filename)asf_obj:
username=json.load(f_obj)
exceptFileNotFoundError:
returnNone
else:
returnusername
defget_new_username():
"""提示用户输入用户名"""
username=input('Whatisyourname?')
filename="username.json"
withopen(filename,'w')asf_obj:
json.dump(username,f_obj)
returnusername
defgreet_user():
"""问候用户，并指出其名字"""
username=get_stored_username()
ifusername:
print("Welcomeback,"+username+"!")
else:
username=get_new_username()
print("We'llrememberyouwhenyoucomeback,"+username+"!")
greet_user()