Python：加載CSV，第一列作為行名，第一行作為列名

Question

我想使用Python2.7加載CSV文件，其中第一行包含列名，第一列包含行名。

我的CSV文件如下所示。

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

我不知道如何用numpy或pandas來做。 有人可以啟發我嗎？ 謝謝 ！

Answer 1

您可以將read_csv與分隔符s\\+ -任意空格一起使用：

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

文件

Answer 2

您可以將pd.read_csv與正則表達式分隔符\\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

或者您可以delim_whitespace使用delim_whitespace參數：

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