python yaml 读写多个列表

'''下面代码完成2个功能

1、写list1、list2进入文件1.yaml

2、从文件1.yaml读取并显示

'''

import yaml

import os

import sys

#获取当前py文件目录。

mypath2=sys.path[0]

#写list1、list2进入1.yaml，与当前py文件同目录--------------------------

f = open(mypath2+'/1.yaml', 'r+', encoding='utf-8')

list1=[1,2,3,4]

list2=[12,3,4,5,6]

yaml.dump_all([list1,list2],f)

f.close()

#读取1.yaml中的两个列表------------------------------------------------------

f = open(mypath2+'/1.yaml', 'r+', encoding='utf-8')

cfg = f.read()

ds = yaml.load_all(cfg)

#ds是所有读取数据的迭代器

for i in ds:

print(i) #输出2个列表