码迷,mamicode.com
首页 > 编程语言 > 详细

【Python】存储数据

时间:2017-11-27 00:09:53      阅读:155      评论:0      收藏:0      [点我收藏+]

标签:tps   welcome   丢失   简单   读取数据   代码   传递   java   isp   

很多程序都要求用户输入某种信息,如让用户存储游戏首选项或者提供可视化数据,不管专注什么,程序都要将数据进行存储,那么如何存储呢?

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

使用json.dump()和json.loan()

1.josn.dump()存储数据

语法:json.dump(存储数据,文件名)

import json
numbers = [1,2,3,4,5,6]
file_name = TestDump.json
with open(file_name,w) as file_obj:
    json.dump(numbers,file_obj)

我们可以在文件夹中看到文件

技术分享图片

2.json.laod()读取数据

语法:json.load(文件名)

下面是我们读取数据文件TestDump.json内容的程序

import json
file_name = TestDump.json
with open(file_name) as file_obj:
    numbers = json.load(file_obj)
print(numbers)

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

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

对于用户生成的数据,使用json保存他们大有裨益,因为如果不以某种方式进行存储,等程序停止运行时用户的数据将会丢失。

import json
username = input("what is your name?")

filename = username.json
with open(filename,w) as file_obj:
    json.dump(username,file_obj)
    print("we will remeber when you come back," + username+"!")

运行结果:

技术分享图片

技术分享图片

我们提示输入用户名,并将其存储在一个变量中,接下来,我们又调用json.dump(),并将用户名和一个文件对象传递给他,从而将用户名存储在文件中。

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

import json
file_name = username.json
with open(file_name) as f_obj:
    username = json.load(f_obj)
    print("Welcome back,"+username+"!")

运行结果:

技术分享图片

我们将两个程序合并到一个程序中。在这个程序运行时,我们将尝试从文件username.json中获取用户名,因此我们首先编写一个尝试恢复用户名的try代码块。当文件不存在的时候,我们就在except中提示输入用户名,并将其存储到username.json文件中,以便程序再次运行时获取它。

import json
#如果以前存储了用户名,就加载它,否则,就提示用户输入用户名
file_name = username.json
try:
    with open(file_name) as f_obj:
        username = json.load(f_obj)
except FileNotFoundError:
    username = input("what is your name?")
    with open(file_name,w) as f_obj:
        json.dump(username,f_obj)
        print("we‘ll remeber you when you come back,"+username+"!")
else:
    print("Welcome back,"+username+"!")

这个程序首次运行输出如下:

技术分享图片

否则,输出如下:

技术分享图片

重构

你经常会遇到这样的情况:代码能够正确运行,但可做进一步的改进,将代码划分为一系列完成具体工作的函数,这个过程称为重构

import json
def greet_user():
    ‘‘‘问候用户,并指出其名字‘‘‘
    file_name = username.json
    try:
        with open(file_name) as f_obj:
            username = json.load(f_obj)
    except FileNotFoundError:
        username  = input("what is your name?")
        with open(file_name,w) as f_obj:
            json.dump(username,f_obj)
            print("we‘ll remeber you when you come back,"+username+"!")
    else:
       print("Welcome back,"+username+"!")
#调用函数
greet_user()

或许有人觉得,上面的程序执行的任务太多了,我们不想它执行那么多任务,我们首先将获取存储的用户名的代码移到另外一个函数中:

【提取存在文件时执行操作】

import json
def  get_stored_username():
    ‘‘‘如果存储了用户名,就获取它‘‘‘
    file_name = username.json
    try:
        with open(file_name) as f_obj:
            username = json.load(f_obj)
    except FileNotFoundError:
        return None
    else:
       return username
def greet_user():
    ‘‘‘问候用户,并指出其名字‘‘‘
    username = get_stored_username()
    if username:
        print("Welcome back," + username + "!")
    else:
        username = input("what is your name?")
        file_name = username.json
        with open(file_name, w) as f_obj:
            json.dump(username, f_obj)
            print("we‘ll remeber you when you come back," + username + "!")
#调用函数
greet_user()

新增的函数get_stored_username()目标很明确,如果文件存储了用户名,这个函数就获取并返回它,如果文件不存在,这个函数就返回None,这是一种不错的方法:函数要么返回预期的值,要么返回None.

【提取没有存储用户名时提示用户输入的操作】

上面的程序还可以继续优化,将greet_user()中的另外一块代码提取出来:将没有存储用户名时提示用户输入的代码放在一个独立的函数中

import json
def  get_stored_username():
    ‘‘‘如果存储了用户名,就获取它‘‘‘
    file_name = username.json
    try:
        with open(file_name) as f_obj:
            username = json.load(f_obj)
    except FileNotFoundError:
        return None
    else:
       return username
def get_new_username():
    ‘‘‘提示用户输入用户名‘‘‘
    username = input("what is your name?")
    file_name = username.json
    with open(file_name,w) as f_obj:
        json.dump(username,f_obj)
        return username
def greet_user():
    ‘‘‘问候用户,并指出其名字‘‘‘
    username = get_stored_username()
    if username:
        print("Welcome back," + username + "!")
    else:
        username = get_new_username()
        print("we‘ll remeber you when you come back," + username + "!")
#调用函数
greet_user()

这个版本是最终版本了,每个函数都执行单一而清晰的任务。

【Python】存储数据

标签:tps   welcome   丢失   简单   读取数据   代码   传递   java   isp   

原文地址:http://www.cnblogs.com/OliverQin/p/7900967.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!