Sum two files column by column in python

Question

I have two files of exactly the same size and same number of columns. I want to add the i^th column of the 1st file to the i^th column of the 2nd file. Is their a neat way for doing this with python?

file1
a a a a a a
a a a a a a
a a a a a a

file2
b b b b b b
b b b b b b
b b b b b b

I want:

(a+b) (a+b) (a+b) (a+b) (a+b) (a+b)
(a+b) (a+b) (a+b) (a+b) (a+b) (a+b)
(a+b) (a+b) (a+b) (a+b) (a+b) (a+b)

EDIT: The above is just a simplification of a more complicated problem of mine. Each file has thousands of rows and I have many files (~100) to perform this kind of operation on.

Please show us what you have tried.

sshashank124
– sshashank124

2014-05-15 11:53:42 +00:00
Commented May 15, 2014 at 11:53 — sshashank124
– sshashank124, Commented May 15, 2014 at 11:53

Messa · Accepted Answer · 2014-05-15 12:02:32Z

1

with open("file1") as f1:
    with open("file2") as f2:
        for line1, line2 in zip(f1, f2):
            items1 = line1.split()
            items2 = line2.split()
            sums = ["({}+{})".format(i1, i2) for i1, i2 in zip(items1, items2)]
            print(" ".join(sums))

answered May 15, 2014 at 12:02

Messa

25.4k10 gold badges77 silver badges101 bronze badges

Sign up to request clarification or add additional context in comments.

Comments

Community · Accepted Answer · 2017-05-23 12:21:25Z

1

pandas DataFrame can be a good choice for such operation. It allows making operation on whole data frames(matrices) e.g df_one.add(df_two)

1 steep read data from files into data frames: http://pandas.pydata.org/pandas-docs/version/0.13.1/generated/pandas.DataFrame.from_csv.html (example: http://www.econpy.org/tutorials/general/csv-pandas-dataframe)

2 add two data frames as shown in this SO answear: Adding two pandas dataframes

edited May 23, 2017 at 12:21

CommunityBot

11 silver badge

answered May 15, 2014 at 11:56

andilabs

23.5k14 gold badges126 silver badges159 bronze badges

1 Comment

andilabs Over a year ago

If needed install pip install pandas

Vijay Kumar · Accepted Answer · 2014-05-15 12:07:22Z

0

i think this will help you

with open("file1") as a, open("file2") as b:
    x = [[int(i) for i in u.split()] for u in a.readlines()]
    y = [[int(i) for i in v.split()] for v in b.readlines()]
    n = len(x)
    m = len(x[0])
    ans = ""
    for i in xrange(n):
        for j in xrange(m):
            ans += str(x[i][j]+y[i][j]) + " "
        print ans[:-1]
        ans = ""

answered May 15, 2014 at 12:07

Vijay Kumar

3451 silver badge7 bronze badges

Comments

Saullo G. P. Castro · Accepted Answer · 2014-05-15 12:26:45Z

0

You can use numpy.loadtxt():

a = np.loadtxt('a.txt', dtype=object)
b = np.loadtxt('b.txt', dtype=object)

which will accept the element-wise string concatenation that you want, and even more:

print('('+a+'+'+b+')')
#array([['(a+b)', '(a+b)', '(a+b)', '(a+b)'],
#       ['(a+b)', '(a+b)', '(a+b)', '(a+b)'],
#       ['(a+b)', '(a+b)', '(a+b)', '(a+b)']], dtype=object)

print(a+b)
#array([['ab', 'ab', 'ab', 'ab'],
#       ['ab', 'ab', 'ab', 'ab'],
#       ['ab', 'ab', 'ab', 'ab']], dtype=object)

 print(3*a)
#array([['aaa', 'aaa', 'aaa', 'aaa'],
#       ['aaa', 'aaa', 'aaa', 'aaa'],
#       ['aaa', 'aaa', 'aaa', 'aaa']], dtype=object)

edited May 15, 2014 at 12:26

answered May 15, 2014 at 12:14

Saullo G. P. Castro

59.4k28 gold badges191 silver badges244 bronze badges

Collectives™ on Stack Overflow

Sum two files column by column in python

4 Answers 4

Comments

1 Comment

Comments

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

4 Answers 4

Comments

1 Comment

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related