This is a follow up question to this: Python: Find keywords in a text file from another text file
I want to put the textual data from line.strip to a CSV (or excel) file in 2 columns.
Here is my attempt:
import numpy as np
import pandas as pd
import csv
with open('C:\invoice.txt') as f:
invoice_data = [line.strip() for line in f if line.strip()]
with open('C:\dict.txt') as f:
dict_data = set([line.strip() for line in f if line.strip()])
for i in range(0, len(invoice_data), 2):
if invoice_data[i] in dict_data:
print(invoice_data[i: i + 2])
with open('C:\\Users\\fam_robo1\\Documents\\sample.csv','w') as csvfile:
fieldnames = ['keyword','data']
writer = csv.DictWriter(csvfile, fieldnames=fieldnames)
writer.writeheader()
for i in range(0, len(invoice_data), 2):
writer.writerow ({'keyword':[invoice_data[i]] , 'data':[invoice_data[i+2]] })
csvfile.close()
Any help would be appreciated .
Full Traceback:
Traceback (most recent call last):
File "C:\Users\fam_robo1\Documents\keyword.py", line 20, in <module> writer.writerow ({'keyword':[invoice_data[i]] , 'data':[invoice_data[i+2]] })
IndexError: list index out of range