python一些代码_Python一些代码

自定义with open打开文件

# 是使用上下文管理协议自定义open

class Open(object):

def __init__(self,filepath,mode=‘r‘,encoding=‘utf8‘):

self.filepath=filepath

self.mode=mode

self.encoding=encoding

def __enter__(self):

self.f=open(self.filepath,mode=self.mode,encoding=self.encoding)

return self.f

def __exit__(self, exc_type, exc_val, exc_tb):

print(‘别瞎写‘)

self.f.close()

return True

def __getattr__(self, item):

return getattr(self.f,item)

with Open(‘a.txt‘,‘w‘) as f:

f.write(‘aaa‘)

f.jdhlasufh # 触发异常，照样能写

自定义range

# 自定义range

class MyRange(object):

def __init__(self,start=0,end=None):

self.start=start

self.end=end

def __iter__(self):

return self

def __next__(self):

if self.start==self.end:

raise StopIteration

n=self.start

self.start+=1

return n

for i in MyRange(2,7):

print(i)

自定义栈

# 自定义栈

class MyStack(list):

def is_empty(self):

return len(self)==0

def peek(self):

return self[0-1]

def size(self):

return len(self)

def push(self,item):

return self.append(item)

stackobj=MyStack((1,22,3,5))

stackobj.push(998)

print(stackobj)

print(stackobj.peek())

print(stackobj.size())

print(stackobj.is_empty())

自定义链表

class Node(object):

# 单个节点对象

def __init__(self,length):

self.length=length

self.next=None

def createlink(lst):

# 创建链表

head=Node(0)

for num in lst:

p=Node(num)

p.next=head.next

head.next=p

head.length+=1

return head

# 创建的链表只要有个头节点就可以代表整个链表

def createlinktail(lst):

# 创建尾插法链表

head=Node(0)

tail=head

for num in lst:

p=Node(num)

tail.next=p

tail=p

head.length+=1

return head

def travellink(head):

p=head.next

while p is not None:

print(p.length)

p=p.next

lst=[1,23,44,56,]

head=createlink(lst)

travellink(head)

print(‘---------------‘)

tail_head=createlinktail(lst)

travellink(tail_head)