Pandas: как использовать первый столбец в качестве индекса

Вы можете использовать следующие методы, чтобы использовать первый столбец в качестве индексного столбца в DataFrame pandas:

Способ 1. Используйте первый столбец в качестве индекса при импорте DataFrame.

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

Способ 2. Используйте первый столбец в качестве индекса с существующим DataFrame.

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

Следующие примеры показывают, как использовать каждый метод на практике.

Пример 1. Используйте первый столбец в качестве индекса при импорте DataFrame.

Допустим, у нас есть следующий CSV-файл с именем my_data.csv :

Если мы импортируем CSV-файл без указания индексного столбца, pandas просто создаст индексный столбец с числовыми значениями, начиная с 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

Однако мы можем использовать аргумент index_col , чтобы указать, что первый столбец файла CSV должен использоваться в качестве столбца индекса:

 #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

Обратите внимание, что столбец команды теперь используется в качестве индексного столбца.

Пример 2. Используйте первый столбец в качестве индекса с существующим DataFrame.

Предположим, у нас есть следующий существующий DataFrame 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]})

#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

Мы можем использовать функцию set_index() , чтобы установить столбец команды в качестве столбца индекса:

 #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

Обратите внимание, что столбец команды теперь используется в качестве индексного столбца.

Дополнительные ресурсы

В следующих руководствах объясняется, как выполнять другие распространенные задачи в pandas:

Как выбрать столбцы по индексу в Pandas DataFrame
Как переименовать индекс в Pandas DataFrame
Как удалить столбцы по индексу в Pandas