python进阶:第五章(操作数据)

问题一:如何读写csv数据?

我们获得了雅虎网站平安银行2017年的股市信息,以csv数据格式存储:

Date,Open,High,Low,Close,Volume,Adj Close
2017/4/24,8.97,8.98,8.89,8.93,39499500,8.93
2017/4/21,8.92,8.99,8.9,8.97,32540800,8.97
2017/4/20,8.9,8.94,8.89,8.92,43763000,8.92
2017/4/19,9.03,9.04,8.9,8.91,79966800,8.91
2017/4/18,9.09,9.1,9.05,9.05,33537600,9.05
2017/4/17,9.08,9.11,9.05,9.1,53189200,9.1
2017/4/14,9.11,9.12,9.06,9.08,49050000,9.08
2017/4/13,9.11,9.14,9.1,9.12,35744200,9.12
2017/4/12,9.16,9.17,9.1,9.12,45533600,9.12

现将交易中成交量超过50000000的记录存储到另一个csv文件中。

解决方案:
使用标准库中的csv模块,可以使用其中reader和writer完成csv文件读写。

In [1]: import csv 

In [2]: csv.reader?
Docstring:
csv_reader = reader(iterable [, dialect='excel']
                        [optional keyword args])
    for row in csv_reader:
        process(row)

The "iterable" argument can be any object that returns a line
of input for each iteration, such as a file object or a list.  The
optional "dialect" parameter is discussed below.  The function
also accepts optional keyword arguments which override settings
provided by the dialect.

The returned object is an iterator.  Each iteration returns a row
of the CSV file (which can span multiple input lines).
Type:      builtin_function_or_method

In [3]: csv.writer?
Docstring:
csv_writer = csv.writer(fileobj [, dialect='excel']
                            [optional keyword args])
    for row in sequence:
        csv_writer.writerow(row)

    [or]

    csv_writer = csv.writer(fileobj [, dialect='excel']
                            [optional keyword args])
    csv_writer.writerows(rows)

The "fileobj" argument can be any object that supports the file API.
Type:      builtin_function_or_method

使用writer()和reader()函数,操作csv文件

In [3]: with open('tab.csv','rt') as fin:
   ...:     cin = csv.reader(fin) 
   ...:     values = [ row for row in cin] 
   ...:     

In [4]: values[0] 
Out[4]: ['Date', 'Open', 'High', 'Low', 'Close', 'Volume', 'Adj Close']

In [5]: values = values[1:]

In [6]: values[0][5]
Out[6]: '39499500'

In [7]: with open('tab.csv','rt') as fin:
   ...:     cin = csv.reader(fin)
   ...:     values = [ row for row in cin] 
   ...:     with open('tab2.csv','wt') as fin2:
   ...:         writer = csv.writer(fin2)
   ...:         headers = values[0] 
   ...:         writer.writerow(headers) 
   ...:         for row in values[1:]:
   ...:             if int(row[5]) >= 50000000:
   ...:                 writer.writerow(row) 
   ...:                 

In [8]: cat tab2.csv 
Date,Open,High,Low,Close,Volume,Adj Close
2017/4/19,9.03,9.04,8.9,8.91,79966800,8.91
2017/4/17,9.08,9.11,9.05,9.1,53189200,9.1
2017/4/11,9.17,9.19,9.09,9.15,61243700,9.15
2017/4/7,9.19,9.22,9.17,9.2,51484400,9.2
2017/3/31,9.08,9.18,9.08,9.17,63312100,9.17
......

问题二:如何读写json数据?

问题内容:在python中如何读写json数据?

解决方案:使用标准库中的json模块,其中loads,dumps函数可以完成json数据的读写。

In [1]: import json 

In [2]: json.       
              json.decoder         json.encoder         json.JSONEncoder     json.scanner         
              json.dump            json.JSONDecodeError json.load                                 
              json.dumps           json.JSONDecoder     json.loads

In [2]: l = [1,2,"abc",{"age":13,"name":"Bob"}]

In [3]: json.dumps(l) 
Out[3]: '[1, 2, "abc", {"age": 13, "name": "Bob"}]'

In [4]: d = {'b':None,'a':5,'c':'abc'} 

In [5]: json.dumps(d) 
Out[5]: '{"b": null, "a": 5, "c": "abc"}'

In [6]: json.dumps?
Signature: json.dumps(obj, skipkeys=False, ensure_ascii=True, check_circular=True, allow_nan=True, cls=None, indent=None, separators=None, default=None, sort_keys=False, **kw)

我们发现,在生成的json串中,每个 ','后面都有一个空格,我们可以修改dumps()函数中的separators参数,指定分隔符不带空格。

In [7]: json.dumps(l,separators=[',' , ':']) 
Out[7]: '[1,2,"abc",{"age":13,"name":"Bob"}]'

我们还可以对输出的结果按照键排序,通过参数sort_keys

In [8]: json.dumps(d,sort_keys=True) 
Out[8]: '{"a": 5, "b": null, "c": "abc"}'

下面我们使用loads()函数将json格式转化为python中的数据格式。

In [9]: l2 = json.loads('[1, 2, "abc", {"age": 13, "name": "Bob"}]')

In [10]: l2
Out[10]: [1, 2, 'abc', {'age': 13, 'name': 'Bob'}]

In [11]: l2[0] 
Out[11]: 1

In [12]: d2 = json.loads('{"b": null, "a": 5, "c": "abc"}')  

In [13]: d2 
Out[13]: {'a': 5, 'b': None, 'c': 'abc'}

函数load()和dump()操作的是文件:

In [14]: with open('demo.json','wt') as f:
    ...:     json.dump(l,f) 
    ...:     

In [15]: cat demo.json
[1, 2, "abc", {"age": 13, "name": "Bob"}]

问题三:如何解析简单的xml文档?

问题内容:python中如何解析xml文档?
解决方案:使用标准库中的xml.etree.ElementTree,其中的parse函数可以解析xml文档。

In [1]: from xml.etree.ElementTree import parse 

In [2]: parse?
Signature: parse(source, parser=None)
Docstring:
Parse XML document into element tree.

*source* is a filename or file object containing XML data,
*parser* is an optional parser instance defaulting to XMLParser.

Return an ElementTree instance.
File:      /usr/local/lib/python3.5/xml/etree/ElementTree.py
Type:      function

In [3]: f = open("demo.xml")

In [4]: et = parse(f) 

In [5]: root = et.getroot()

In [6]: root
Out[6]: <Element 'breakfast_menu' at 0x7f788b23d548>

In [7]: root.tag
Out[7]: 'breakfast_menu'

In [8]: root.attrib
Out[8]: {}

In [9]: root.              
              root.append      root.findall     root.getiterator root.iterfind    root.remove      
              root.clear       root.findtext    root.insert      root.itertext    root.set         
              root.extend      root.get         root.items       root.keys                         
              root.find        root.getchildren root.iter        root.makeelement

我们根据上面的函数,可以操作xml数据。
find()和findall()方法只能查找当前节点的字节点,不过当我们使用iter()方法的时候,可以遍历所有子节点。

问题四:如何构建xml文档?

如何将其它文档转化为xml文旦。
解决方案:使用标准库中的xml.etree.ElementTree,构建ElementTree,使用write方法写入文件。

In [1]: from xml.etree.ElementTree import Element,ElementTree 

In [2]: from xml.etree.ElementTree import tostring 

tostring可以查看转化为xml格式后的样式

添加跟节点
In [3]: e = Element('Data') 

In [4]: e.tag
Out[4]: 'Data'

为节点设置属性
In [5]: e.set('name','abc') 

In [6]: tostring(e) 
Out[6]: b'<Data name="abc" />'

为节点设置文本值
In [7]: e.text = '123' 

In [8]: tostring(e) 
Out[8]: b'<Data name="abc">123</Data>'

新增两个节点
In [9]: e2 = Element('Row') 

In [10]: e3 = Element('Open') 

In [11]: e3.text = '8.80' 

将e3设置为e2的子节点
In [12]: e2.append(e3) 

In [13]: tostring(e2) 
Out[13]: b'<Row><Open>8.80</Open></Row>'

In [14]: e.text = None 

将e2设置为e的字节点
In [15]: e.append(e2) 

In [16]: tostring(e) 
Out[16]: b'<Data name="abc"><Row><Open>8.80</Open></Row></Data>'

创建ElementTree元素用于写入文件,传入的参数是根元素
In [17]: et = ElementTree(e) 

使用write()函数,将字符串写入到文件
In [18]: et.write('demo2.xml') 

In [19]: cat demo2.xml
<Data name="abc"><Row><Open>8.80</Open></Row></Data>

我们将之前的股票交易转化为xml格式:

In [50]: from xml.etree.ElementTree import Element,ElementTree 

In [51]: import csv 

该函数将输出的xml格式规范化
In [52]: def pretty(e,level=0):
    ...:     if len(e) > 0:
    ...:         e.text = '\n' + '\t' * (level + 1) 
    ...:         for child in e :
    ...:             pretty(child,level + 1)
    ...:         child.tail = child.tail[:-1] 
    ...:     e.tail = '\n' + '\t' * level
    ...:     

In [53]: def csvToXml(fname): 
    ...:     with open(fname,'rt') as f :
    ...:         reader = csv.reader(f) 
    ...:         values = [ row for row in reader]
    ...:         headers =  values[0] 
    ...:         root = Element('Data') 
    ...:         for row in values[1:]:
    ...:             eRow = Element('Row')
    ...:             root.append(eRow)
    ...:             for tag , text in zip(headers,row):
    ...:                 e = Element(tag) 
    ...:                 e.text = text 
    ...:                 eRow.append(e)
    ...:         pretty(root) 
    ...:         return ElementTree(root) 
    ...:     

In [54]: et = csvToXml('tab.csv')

In [55]: et.write("pay.xml") 

问题五:如何读写excel文件

问题内容:数据格式为xls,xlsx,是一种常用的电子表格。小学某班级成绩,记录在excel文件中:
姓名 语文 数学 外语
李磊 95 99 96
韩梅 98 100 93
张峰 94 95 95
......
利用python读写excel,添加“总分”列,计算每人总分。

解决方案:
使用第三方库xlrd和xlwt,这两个库分别用于excel读和写。

之后有时间补充该内容:
相关之前的文章<a>http://www.jianshu.com/p/32d6b528d5c5</a>

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 157,298评论 4 360
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 66,701评论 1 290
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 107,078评论 0 237
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 43,687评论 0 202
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 52,018评论 3 286
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 40,410评论 1 211
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 31,729评论 2 310
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 30,412评论 0 194
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 34,124评论 1 239
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 30,379评论 2 242
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 31,903评论 1 257
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 28,268评论 2 251
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 32,894评论 3 233
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 26,014评论 0 8
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 26,770评论 0 192
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 35,435评论 2 269
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 35,312评论 2 260

推荐阅读更多精彩内容