使用 python 创建 yaml 文件 - 列表格式

Question

我有一个 python 脚本，它使用yaml.dump(yaml_dict, file_path)输出 yaml 文件

output 的格式如下：

name: joe
type: builder
tables:
- table_name_1:
    exclusion:
    - column_a
    - column_b
    site: location_c
- table_name_2:
    exclusion:
    - column_d
    site: location_b

这与我希望 output 的外观非常接近，但在表名的格式方面略有不同。 这就是我要的：

name: joe
type: builder
tables:
  table_name_1:
    exclusion:
    - column_a
    - column_b
    site: location_c
  table_name_2:
    exclusion:
    - column_d
    site: location_b

也就是说，我不想在表格列表之前使用连字符，而只需要通过空格进行缩进。

我用来将表添加到yaml_dict的代码如下：

table_list = df['table'].tolist()
    for table in table_list:
        table_config = {'site': df.loc[df['table'] == table, 'site'].item()}
        exclusion = df.loc[df['table'] == table, 'exclusion'].item()
        er_list = exclusion.replace(' ', '').split(",")
        table_config['exclusion'] = er_list
        table_dict = {table: table_config}
        yaml_dict['tables'].append(table_dict)

我很高兴将连字符保留在排除列表中，但我不希望它们出现在表格列表中。 我能做些什么来删除它们吗？

Answer 1

破折号表示您正在输出一个列表。 相反，你需要一个字典。

假设你初始化它是这样的：

yaml_dict['tables'] = {}

然后在for循环中：

yaml_dict['tables'].update(table_dict)