在csv文件中添加索引列

时间:2017-04-12 09:16:51

标签: python csv pandas dataframe

我有以下样本进行转换。连接几个csv文件后,我将每行0的索引保持到每个文件中文件的最后一行,如下所示。

    Column_1    column2
0   m            4
1   n            3
2   4            6
3   t            8
0   h            8
1   4            7
2   kl           8
3   m            4
4   bv           5
5   n            8

现在我想在开头索引文件时添加另一列。

        Column_1    column2
0   0   m           4
1   1   n           3
2   2   4           6
3   3   t           8
4   0   h           8
5   1   4           7
6   2   kl          8
7   3   m           4
8   4   bv          5
9   5   n           8

1 个答案:

答案 0 :(得分:3)

最简单的是MultiIndex.from_arrays numpy.arangerange

print (np.arange(len(df.index)))
[0 1 2 3 4 5 6 7 8 9]

n = ['a','b']
df.index = pd.MultiIndex.from_arrays([np.arange(len(df.index)), df.index], names= n)
print (df)
    Column_1  column2
a b                  
0 0        m        4
1 1        n        3
2 2        4        6
3 3        t        8
4 0        h        8
5 1        4        7
6 2       kl        8
7 3        m        4
8 4       bv        5
9 5        n        8
n = ['a','b']
df.index = pd.MultiIndex.from_arrays([range(len(df.index)), df.index], names= n)
print (df)
    Column_1  column2
a b                  
0 0        m        4
1 1        n        3
2 2        4        6
3 3        t        8
4 0        h        8
5 1        4        7
6 2       kl        8
7 3        m        4
8 4       bv        5
9 5        n        8

如果不需要索引名称,只需指定:

df.index = [np.arange(len(df.index)), df.index]
print (df)
    Column_1  column2
0 0        m        4
1 1        n        3
2 2        4        6
3 3        t        8
4 0        h        8
5 1        4        7
6 2       kl        8
7 3        m        4
8 4       bv        5
9 5        n        8