Python：加载CSV，第一列作为行名，第一行作为列名

Question

I want to load a CSV file using Python2.7, in which the first row contains column names and the first column contains row names.

My CSV file looks like beneath.

  A  B  C  D
a 1. 2. 3. 4. 
b 5. 6. 7. 8. 

I don't know how to do that with numpy or pandas. Can someone enlighten me ? Thanks !

Answer 1

You can use read_csv with separator s\\+ - arbitrary whitespace:

import pandas as pd
import io

temp=u"""A B C D
a 1. 2. 3. 4.
b 5. 6. 7. 8."""
#after testing replace io.StringIO(temp) to filename
df = pd.read_csv(io.StringIO(temp), sep="\s+")
print df
     A    B    C    D
a  1.0  2.0  3.0  4.0
b  5.0  6.0  7.0  8.0

Docs

Answer 2

You could use pd.read_csv with regex separator \\s+ :

import pandas as pd

In [4]: pd.read_csv('file.csv', sep='\s+')
Out[4]:
     A    B    C    D
a  1.0  2.0  3.0  4.0
b  5.0  6.0  7.0  8.0

Or you could use delim_whitespace argument for that:

In [5]: pd.read_csv('file.csv', delim_whitespace=True)
Out[5]:
     A    B    C    D
a  1.0  2.0  3.0  4.0
b  5.0  6.0  7.0  8.0