how to write user-defined column names to a csv file using pandas?

Question

Is is possible to write my own column names when writing a pivot table to a csv file?

import pandas as pd

c1 = ['a','a','a','a','b','b','b','b']
c2 = ['x','x','y','y','x','x','y','y']
c3 = [1, 2, 1, 2, 1, 2, 1, 2]
val = [85, 47, 29, 93, 15, 21, 65, 16]

df = pd.DataFrame({'c1':c1, 'c2':c2, 'c3':c3, 'val':val})

ptable = pd.pivot_table(data=df, cols=['c2','c3'], rows='c1')

I tried to use the header parameter:

ptable.to_csv('test.csv', header=['n1','n2','n3','n4'])

but the column names weren't changed...

unutbu · Accepted Answer · 2014-02-06 23:05:40Z

3

Here is a work-around: Change the columns of ptable before calling to_csv:

ptable = pd.pivot_table(data=df, cols=['c2','c3'], rows='c1')
ptable.columns = ['n1','n2','n3','n4']
ptable.to_csv('/tmp/test.csv')

answered Feb 6, 2014 at 23:05

unutbu

886k197 gold badges1.9k silver badges1.7k bronze badges

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

Comments

EdChum · Accepted Answer · 2014-02-06 23:05:56Z

2

Just rename before writing:

ptable.columns=['n1','n2','n3','n4']
ptable.to_csv(r'c:\data\test.csv')

It should in fact work passing the list for the header parameter, not sure why, could be a bug

answered Feb 6, 2014 at 23:05

EdChum

397k204 gold badges837 silver badges583 bronze badges

Collectives™ on Stack Overflow

how to write user-defined column names to a csv file using pandas?

2 Answers 2

Comments

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Related