dataframe,从列表中设置索引

Question

从列表创建数据帧时,是否可以将索引设置为其中一个值？

import pandas as pd

tmp = [['a', 'a1'], ['b',' b1']]

df = pd.DataFrame(tmp, columns=["First", "Second"])

        First  Second
0          a   a1
1          b   b1

以及我希望它看起来如何:

        First  Second
a          a   a1
b          b   b1

Answer 1

set_axis

\n

要将任意值设置为索引，最佳实践是使用set_axis：

\n

df = df.set_axis([\'idx1\', \'idx2\'])\n\n#       First  Second\n# idx1      a      a1\n# idx2      b      b1\n

\n\n

set_index（列表与数组）

\n

也可以将任意值传递给set_index，但请注意传递列表与数组之间的区别：

\n

\n

• list \xe2\x80\x94set_index将这些列指定为索引：

  \n

  df.set_index([\'First\', \'First\'])\n\n#              Second\n# First First        \n# a     a          a1\n# b     b          b1\n

  \n
\n

• array (Series/Index/ndarray) \xe2\x80\x94set_index将这些值指定为索引：

  \n

  df = df.set_index(pd.Series([\'First\', \'First\']))\n\n#        First  Second\n# First      a      a1\n# First      b      b1\n

  \n

  _{请注意，将数组传递给在开发人员中非常有争议set_index，甚至可能被弃用。}

  \n
\n

\n\n

为什么不df.index直接修改呢？

\n

直接修改属性很好，也经常使用，但使用方法有其优点：

\n

\n

• 方法提供更好的错误检查，例如：

  \n

  df = df.set_axis([\'idx1\', \'idx2\', \'idx3\'])\n\n# ValueError: Length mismatch: Expected axis has 2 elements, new values have 3 elements\n

  \n

  df.index = [\'idx1\', \'idx2\', \'idx3\']\n\n# No error despite length mismatch\n

  \n
\n

• 方法可以链接起来，例如：

  \n

  df.some_method().set_axis([\'idx1\', \'idx2\']).another_method()\n

  \n
\n

\n

Answer 2

>>> pd.DataFrame(tmp, columns=["First", "Second"]).set_index('First', drop=False)
      First Second
First             
a         a     a1
b         b     b1

Answer 3

在将其分配给索引之前将其更改为列表

df.index = list(df["First"])

Answer 4

如果您不想要索引名称:

df = pd.DataFrame(tmp, columns=["First", "Second"], index=[i[0] for i in tmp])

结果:

  First Second
a     a     a1
b     b     b1