如何使用python 2.7导出带有标题的CSV文件

Question

I'm trying to figure out how to export the results of my script to a CSV file with python 2.7. The CSV file should contain two columns:

The first column should contain the URL results and I would like to give this column a name. The second column should contain the print result keyword found or keyword NOT found (as seen after the first and second print function in my code). I would like to name the second column as well.

My code as of now:

import urllib2

keyword = ['viewport']


with open('top1m-edited.csv', 'w') as f:
    # Write out your column headers
    f.write(','.join(['column1header', 'column2header']))

with open('top1m-edited.csv') as f:
    for line in f:
        strdomain = line.strip()
        if '.nl' in strdomain:
            try:
                req = urllib2.Request(strdomain.strip())
                response = urllib2.urlopen(req)
                html_content = response.read()

                for searchstring in keyword:
                    if searchstring.lower() in str(html_content).lower():
                        f.write(','.join([strdomain, 'keyword found']) + '\n')
                    else:
                        f.write(','.join([strdomain, 'keyword NOT found']) + '\n')
                        print (strdomain, 'keyword NOT found')

f.close()

I'm getting IndentationError: unexpected unindent

So what should I adjust to make this work?

Answer 1

You can use the ','.join() method to convert a list into a string with a comma separator.

with open('my_file.csv', 'w') as f:
    # Write out your column headers
    f.write(','.join(['column1header', 'column2header']))

    # Replace your for loop with this to write to file instead of stdout
    for searchstring in keyword:
        if searchstring.lower() in str(html_content).lower():
            f.write(','.join([strdomain, 'keyword found']) + '\n')
        else:
            f.write(','.join([strdomain, 'keyword NOT found']) + '\n')
            print (strdomain, 'keyword NOT found')