如何将 pandas 索引转换为列表(附示例)


您可以使用两种方法之一将 pandas DataFrame 的索引转换为列表:

方法一:使用list()

 index_list = list( df.index.values )

方法2:使用tolist()

 index_list = df. index . values . tolist ()

两种方法都会返回完全相同的结果,但第二种方法在极大的 DataFrame 上往往会更快。

以下示例展示了如何将每种方法与以下 pandas DataFrame 一起使用:

 import pandas as pd

#createDataFrame
df = pd. DataFrame ({' team ': ['A', 'B', 'C', 'D', 'E', 'F', 'G', 'H'],
                   ' points ': [18, 22, 19, 14, 14, 11, 20, 28],
                   ' assists ': [5, 7, 7, 9, 12, 9, 9, 4],
                   ' rebounds ': [11, 8, 10, 6, 6, 5, 9, 12]})

#view DataFrame
df

	team points assists rebounds
0 A 18 5 11
1 B 22 7 8
2 C 19 7 10
3 D 14 9 6
4 E 14 12 6
5 F 11 9 5
6 G 20 9 9
7:28 4 12

方法一:使用list()

下面的代码展示了如何使用list()快速将pandas DataFrame的索引转换为列表:

 #convert index to list
index_list = list( df.index.values )

#view list
index_list

[0, 1, 2, 3, 4, 5, 6, 7]

请注意,该列表包含 DataFrame 索引中的原始值。

我们还可以使用type()来验证结果是否是一个列表:

 #check object type
type (index_list)

list

方法2:使用tolist()

下面的代码展示了如何使用list()快速将pandas DataFrame的索引转换为列表:

 #convert index to list
index_list = df. index . values.tolist ()

#view list
index_list

[0, 1, 2, 3, 4, 5, 6, 7]

同样,该列表包含 DataFrame 索引中的原始值。

我们还可以使用type()来验证结果是否是一个列表:

 #check object type
type (index_list)

list

其他资源

以下教程解释了如何在 pandas 中执行其他常见操作:

如何修改Pandas中的一个或多个索引值
如何在 Pandas 中将索引转换为列
如何重置 Pandas 中的索引

添加评论

您的电子邮箱地址不会被公开。 必填项已用*标注