我有一个csv文件,其中包含数据,其中标头包含键,单元格包含值.我想使用
python从csv文件的内容创建一个yaml文件.
我创建了一个K:V对的字典;但是,我试图让K:V对进入yaml文件.
yaml的结构必须是:
key1: value1
key2: value2
key3:
- key4: value4
key5: {key6: [value6]
key7: value7
key8: value8
key9: value9
-
-
---
如果我要手动创建这些,我将拥有超过1000个YAML,因此它非常耗时且不切实际.
我正在寻找您更有经验的人可能有的任何想法.
我真的希望输出迭代字典来创建一个巨大的YAML列表,如下所示:
key1: value1
key2: value2
key3:
- key4: value4
key5: {key6: [value6]
key7: value7
key8: value8
key9: value9
-
-
---
key1: value1
key2: value2
key3:
- key4: value4
key5: {key6: [value6]
key7: value7
key8: value8
key9: value9
-
-
---
key1: value1
key2: value2
key3:
- key4: value4
key5: {key6: [value6]
key7: value7
key8: value8
key9: value9
-
-
---
key1: value1
key2: value2
key3:
- key4: value4
key5: {key6: [value6]
key7: value7
key8: value8
key9: value9
-
-
---
示例代码:
import csv
import yaml
def csv_dict_list(variables_file) :
reader=csv.DictReader(open(variables_file, 'r'))
dict_list = []
for line in reader:
dict_list.append(line)
return dict_list
yaml_values = csv_dict_list(sys.argv[1])
无论我在此之后尝试什么,我都无法使用yaml.load()或yaml.load_all()获得所需的输出.
最佳答案 首先,您应该使用dump()或dump_all(),因为您要编写YAML而不是使用load().
您还应该知道CSV阅读器确实在Python 2.7上返回了不同的内容,例如在Python 3.6上:首先从csv_dict_list获取dict列表,在第二个上获得OrderedDict列表).这本身不会有问题,但PyYAML会转储一个带有键的dict,以及带有标签的ordereddict.
您提议的YAML也无效,因为行中的流样式映射:
key5: {key6: [value6]
在文档结束之前没有以}结尾,你也不能拥有:
key9: value9
-
-
使用:
key9: value9
key10:
-
-
要么
key9:
- value9
-
或者类似的东西(也没有相同的Python数据结构同时拥有一个值和一个列表,因此即使在Python中也不能创建类似的东西).
PyYAML还缺少对缩进块样式序列的支持.如果你这样做:
import yaml
print(yaml.dump(dict(x=[dict(a=1, b=2)]), indent=4))
输出仍然是齐平左:
x:
- {a: 1, b: 2}
为了防止在使用PyYAML时遇到的所有这些问题,并且为了规避Python版本的差异,我建议你使用ruamel.yaml(免责声明:我是该软件包的作者),以及以下代码:
import sys
import csv
import ruamel.yaml
Dict = ruamel.yaml.comments.CommentedMap
def csv_dict_list(variables_file) :
reader=csv.reader(open(variables_file, 'r'))
key_list = None
dict_list = []
for line in reader:
if key_list is None:
key_list = line
continue
d = Dict()
for idx, v in enumerate(line):
k = key_list[idx]
# special handling of key3/key4/key5/key6
if k == key_list[2]:
d[k] = []
elif k == key_list[3]:
d[key_list[2]].append(Dict([(k, v)]))
elif k == key_list[4]:
d[key_list[2]][0][k] = dt = Dict()
dt.fa.set_flow_style()
elif k == key_list[5]:
d[key_list[2]][0][key_list[4]][k] = [v]
else:
d[k] = v
dict_list.append(d)
return dict_list
data = csv_dict_list('test.csv')
yaml = ruamel.yaml.YAML()
yaml.indent(sequence=4, offset=2)
yaml.dump_all(data, sys.stdout)
使用test.csv:
key1,key2,key3,key4,key5,key6,key7,key8,key9
value_a1,value_a2,value_a3,value_a4,value_a5,value_a6,value_a7,value_a8,value_a9
value_b1,value_b2,value_b3,value_b4,value_b5,value_b6,value_b7,value_b8,value_b9
这给了:
key1: value_a1
key2: value_a2
key3:
- key4: value_a4
key5: {key6: [value_a6]}
key7: value_a7
key8: value_a8
key9: value_a9
---
key1: value_b1
key2: value_b2
key3:
- key4: value_b4
key5: {key6: [value_b6]}
key7: value_b7
key8: value_b8
key9: value_b9
在Python 2.7和Python 3.6上