Including filename when reading a file in Python

Question

I am trying to read a file in Python using the below code:

with open(file) as fp:
    data = fp.read()

The above script works well. I however am trying to have the filename included in the output of data

How could I have that included.

Edit:

Adding a sample output with the expected result

Current Output:

col1,col2,col3
1,Prod_A,10
2,Prod_B,5

Expected output:

col1,col2,col3,filename
1,Prod_A,10,sales.csv
2,Prod_B,5,sales.csv

`fp.name` contains the name of the file. is that what you need? — hiro protagonist, Mar 28 '19 at 07:58
Well that's awfully unclear. Assuming the file name is `"foo.txt"` and the file content is `"bar"`, you would want `data` to be... what? — Aran-Fey, Mar 28 '19 at 08:00
`file.split('/')[-1]` (or with'\') gives you the name before you even open it — Nenri, Mar 28 '19 at 08:05
@hiroprotagonist, I have edited my post with some sample output. hope this helps. — scott martin, Mar 28 '19 at 08:52
@Aran-Fey, trying to include foo as one of the columns in the output. Added a sample output for reference in my initial post. — scott martin, Mar 28 '19 at 08:55
Suspected dup: [How to add a new column to a CSV file?](https://stackoverflow.com/questions/11070527/how-to-add-a-new-column-to-a-csv-file) — martineau, Mar 28 '19 at 09:12

not link · Answer 1 · 2019-04-01T16:02:09.660

0

It might be something as simple as

data = {}

with open("test.txt") as fp:
   data['name'] = fp.name
   data['content'] = fp.read()

the output would look something like

{'name': 'test.txt',
 'content': '124512451245124512451245124512451245124512451245\n'}

But it really depends on what you want the structure of data to be.

Edit: based on the feedback, the example above can be rewritten with a list as

data = []

with open("test.txt") as fp:
   data.append(fp.name)
   data.append(fp.read())

the output would look like:

['test.txt', '124512451245124512451245124512451245124512451245\n']

edited Apr 01 '19 at 16:02

answered Mar 28 '19 at 08:08

not link

858
1
16
29

trying to achieve this using a list – scott martin Mar 28 '19 at 08:54
@scottmartin does the updated answer address your question? If so, please don't forget to close the question :) – not link Apr 01 '19 at 16:03

Camile · Answer 2 · 2019-03-28T12:27:15.017

0

Read your csv file and add new column:

import csv
import os

m, n = 'sales.csv', 'sales_temp.csv'

with open(m, newline='') as f:
    with open(n, 'w', newline='') as f_temp:
        w = csv.writer(f_temp)

        for r in csv.reader(f):
            w.writerow(r + ['filename']) if r[0] == 'col1' else w.writerow(r + [f.name])

os.remove(m)
os.rename(n, m)

edited Mar 28 '19 at 12:27

answered Mar 28 '19 at 12:06

Camile

107
5

Including filename when reading a file in Python

2 Answers2