如何使用Pandas读取带有行名称的数据框的CSV文件。

Question

我有一个 CSV 文件 (tmp.csv)，内容如下：

        x       y       z
bar     0.55    0.55    0.0
foo     0.3     0.4     0.1
qux     0.0     0.3     5.55

这是使用Pandas创建的方式：

    In [103]: df_dummy 
    Out[103]: 
          x     y     z
    bar  0.55  0.55  0.00
    foo  0.30  0.40  0.10
    qux  0.00  0.30  5.55

   In [104]: df_dummy.to_csv("tmp.csv",sep="\t")   

我的目的是将那个CSV文件读入相同的数据框架。我尝试了以下代码，但并没有得到我想要的结果：

In [108]: pd.io.parsers.read_csv("tmp.csv",sep="\t")
Out[108]: 
  Unnamed: 0     x     y     z
0        bar  0.55  0.55  0.00
1        foo  0.30  0.40  0.10
2        qux  0.00  0.30  5.55

怎么做才是正确的方式？

- pdubois

Answer 1

您可以使用 index_col 参数：

>>> pd.io.parsers.read_csv("tmp.csv",sep="\t",index_col=0)
        x     y     z
bar  0.55  0.55  0.00
foo  0.30  0.40  0.10
qux  0.00  0.30  5.55

- Roman Pekar