Como converter o índice pandas em lista (com exemplos)

Você pode usar um dos dois métodos para converter o índice de um DataFrame do pandas em uma lista:

Método 1: use lista()

 index_list = list( df.index.values )

Método 2: use tolist()

 index_list = df. index . values . tolist ()

Ambos os métodos retornarão exatamente o mesmo resultado, mas o segundo método tenderá a ser mais rápido em DataFrames extremamente grandes.

Os exemplos a seguir mostram como usar cada método com o seguinte DataFrame do pandas:

 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

Método 1: use lista()

O código a seguir mostra como usar list() para converter rapidamente o índice do DataFrame do pandas em uma lista:

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

#view list
index_list

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

Observe que a lista contém os valores originais do índice do DataFrame.

Também podemos usar type() para verificar se o resultado é uma lista:

 #check object type
type (index_list)

list

Método 2: use tolist()

O código a seguir mostra como usar list() para converter rapidamente o índice do DataFrame do pandas em uma lista:

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

#view list
index_list

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

Novamente, a lista contém os valores originais no índice do DataFrame.

Também podemos usar type() para verificar se o resultado é uma lista:

 #check object type
type (index_list)

list

Recursos adicionais

Os tutoriais a seguir explicam como realizar outras operações comuns em pandas:

Como modificar um ou mais valores de índice no Pandas
Como converter índice em coluna no Pandas
Como redefinir um índice no Pandas