Pandas: como usar a primeira coluna como índice

Você pode usar os seguintes métodos para usar a primeira coluna como uma coluna de índice em um DataFrame do pandas:

Método 1: Use a primeira coluna como índice ao importar DataFrame

 df = pd. read_csv (' my_data.csv ', index_col= 0 )

Método 2: Use a primeira coluna como um índice com um DataFrame existente

 df = df. set_index ([' column1 '])

Os exemplos a seguir mostram como usar cada método na prática.

Exemplo 1: Use a primeira coluna como índice ao importar DataFrame

Digamos que temos o seguinte arquivo CSV chamado my_data.csv :

Se importarmos o arquivo CSV sem especificar uma coluna de índice, o pandas simplesmente criará uma coluna de índice com valores numéricos começando em 0:

 #import CSV file without specifying index column
df = pd. read_csv (' my_data.csv ')

#view DataFrame
print (df)

  team points assists
0 to 18 5
1 B 22 7
2 C 19 7
3 D 14 9
4 E 14 12
5 F 11 9
6 G 20 9
7:28 a.m. 4

No entanto, podemos usar o argumento index_col para especificar que a primeira coluna do arquivo CSV deve ser usada como coluna de índice:

 #import CSV file and specify index column
df = pd. read_csv (' my_data.csv ', index_col= 0 )

#view DataFrame
print (df)

      assist points
team                 
At 18 5
B 22 7
C 19 7
D 14 9
E 14 12
F 11 9
G 20 9
H 28 4

Observe que a coluna team agora é usada como coluna de índice.

Exemplo 2: Use a primeira coluna como um índice com um DataFrame existente

Vamos supor que temos o seguinte DataFrame do pandas existente:

 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]})

#view DataFrame
df

team points assists
0 to 18 5
1 B 22 7
2 C 19 7
3 D 14 9
4 E 14 12
5 F 11 9
6 G 20 9
7:28 a.m. 4

Podemos usar a função set_index() para definir a coluna da equipe como a coluna do índice:

 #set 'team' column as index column
df = df. set_index ([' team '])

#view updated DataFrame
print (df)

      assist points
team                 
At 18 5
B 22 7
C 19 7
D 14 9
E 14 12
F 11 9
G 20 9
H 28 4

Observe que a coluna team agora é usada como coluna de índice.

Recursos adicionais

Os tutoriais a seguir explicam como realizar outras tarefas comuns em pandas:

Como selecionar colunas por índice em um DataFrame do Pandas
Como renomear o índice no Pandas DataFrame
Como deletar colunas por índice no Pandas