python时分复用_python经常复用的程序片段记载

1 有输入参数

import sys, os

if len(sys.argv) < 3:

print 'input invalid'

exit(1)

a = sys.argv[1]

b = sys.argv[2]

if not a.isdigit() or not b.isdigit():

print 'input is not digit'

exit(1)

2 打开文件错误捕捉

try:

f1 = open(str(i))

except IOError, e:

print 'cannot open file ' + str(i)

i += 1

continue

line = f1.readline()

dic = eval(line)

f1.close()

3  用urllib和urllib2发送post请求

body = urllib.urlencode({

'token':token,

'task':'{"TaskName":"testScipt"}'})

returnedReq = urllib2.Request(

url = 'http://www.test.com/',

data = body)

returnedResult = urllib2.urlopen(returnedReq).read()

returnedResult = returnedResult.strip()

print returnedResult

4 操作excel

import xlrd

data = xlrd.open_workbook('test.xlsx')

table = data.sheets()[0] #选择第一个工作区

cols = table.col_values(0) #选择第一列

print len(cols)

5  使用eval()或ast.literal_eval() 出现错误

NameError: name ‘total_count’ is not defined或ValueError: malformed string

比如原始字符串是 str = '{"name":"jgy","age":null}'

转为字典时会发现null不是一个字符串，就当做变量，但是没有定义(python认可的是Null True False 注意第一个字母大写)，所以出错，相应的实际过程中可能还会到 ‘true’ is not defined, ‘false’ is not defined类似的等等。

既然没定义，在前面定义一下就行了，比如自己定义 null = 'null',  true = 'true'..

还有个想法，可以用正则把冒号后面的不是数字也没加引号的value值给加上引号。

6 时间

获取最简洁日期格式：

>>>time.strftime('%Y-%m-%d',time.localtime(time.time()))

'2013-07-04'

获取上一秒时间：

time.strftime('%Y-%m-%d %H:%M:%S',time.localtime(time.time() - 1))

将一个字符串时间转为python时间格式

str1 = '2013-06-03 20:07:41'

str2 = '2013-06-03 20:08:43'

time1= time.strptime(str1,"%Y-%m-%d %H:%M:%S") #time1 is time_struct type

time1= time.strptime(str2,"%Y-%m-%d %H:%M:%S")

#now we wanna to 取得time1和time2的时间差

#需要转为datetime，然后使用datetime的timedelta方法

datetime1 = datetime.datetime(*time1[:6])

datetime2 = datetime.datetime(*time2[:6])

datedelta = datetime2 - datetime1

print datedelta

#datetime.timedelta(0, 62) #后面是秒数

seconds = delta.seconds #获取int型秒数

7 海量数据去重

>>> a = [11,22,33,44,11,22]

>>> b = set(a)

>>> b

set([33, 11, 44, 22])

>>> c = [i for i in b]

>>> c

[33, 11, 44, 22]

8