Convert YAML to CSV File Using Python 3 and pyyaml

0Shares

In this article, I’ll give you the complete source code of yaml to csv python script. It uses Python’s pyyaml module to convert YAML to CSV file.

pip install csv
pip install pyyaml

app.py

import csv
import yaml

fieldnames = ['Name', 'Author', 'Written', 'About', 'Body']

with open('output.csv', 'w', newline='') as f_output:
    csv_output = csv.DictWriter(f_output, fieldnames=fieldnames)
    csv_output.writeheader()

    for filename in ['Test_Co_1.txt', 'Test_Co_2.txt']:
        with open(filename) as f_input:
            data = yaml.safe_load(f_input)

        name = data[0]['Name']

        for entry in data:
            key = next(iter(entry))

            if key.startswith('Note') or key.startswith('Comment'):
                row = {'Name' : name}

                for d in entry[key]:
                    for get in ['Author', 'Written', 'About', 'Body']:
                        try:
                            row[get] = d[get]
                        except KeyError as e:
                            pass

                csv_output.writerow(row)
0Shares

Leave a Comment

This site uses Akismet to reduce spam. Learn how your comment data is processed.