Название или название столбца индекса панд

255

Как я могу получить имя столбца индекса в пандах Python? Вот пример кадра данных:

             Column 1
Index Title          
Apples              1
Oranges             2
Puppies             3
Ducks               4  

Я пытаюсь получить / установить заголовок индекса данных. Вот что я попробовал:

import pandas as pd
data = {'Column 1'     : [1., 2., 3., 4.],
        'Index Title'  : ["Apples", "Oranges", "Puppies", "Ducks"]}
df = pd.DataFrame(data)
df.index = df["Index Title"]
del df["Index Title"]
print df

Кто-нибудь знает, как это сделать?

Радикальный Эдвард
источник

Ответы:

367

Вы можете просто получить / установить индекс через его nameсвойство

In [7]: df.index.name
Out[7]: 'Index Title'

In [8]: df.index.name = 'foo'

In [9]: df.index.name
Out[9]: 'foo'

In [10]: df
Out[10]: 
         Column 1
foo              
Apples          1
Oranges         2
Puppies         3
Ducks           4
Джефф
источник
10
На данный момент (0.16) это не работает. Вернее - он работает, но как только DataFrame изменяется, он удаляет имя индекса.
Петр Мигдаль
11
должно быть возможно указать имя индекса во время создания DataFrame. например pd.DataFrame(values,index={"INDEX_NAME":index_values}). Я не понимаю, почему это не разрешено или не реализовано?
Денфромуфа
1
вы можете напрямую создать индекс, чтобы добавить имя
Джефф
@Jeff, похоже, что вы заметили, что создание индекса сначала (и использование его для индекса и столбцов информационного кадра) является правильным подходом, хотя я согласен с тем, @denfromufaчто он должен принимать dict в качестве параметров для построения из pandas.DataFrame
Diego Aguado
3
Если это мультииндекс, используйте df.index.namesвместо df.index.name.
Яша
75

Ты можешь использовать rename_axis , для удаления установить None:

d = {'Index Title': ['Apples', 'Oranges', 'Puppies', 'Ducks'],'Column 1': [1.0, 2.0, 3.0, 4.0]}
df = pd.DataFrame(d).set_index('Index Title')
print (df)
             Column 1
Index Title          
Apples            1.0
Oranges           2.0
Puppies           3.0
Ducks             4.0

print (df.index.name)
Index Title

print (df.columns.name)
None

Новая функциональность хорошо работает в цепочках методов.

df = df.rename_axis('foo')
print (df)
         Column 1
foo              
Apples        1.0
Oranges       2.0
Puppies       3.0
Ducks         4.0

Вы также можете переименовать имена столбцов с параметром axis :

d = {'Index Title': ['Apples', 'Oranges', 'Puppies', 'Ducks'],'Column 1': [1.0, 2.0, 3.0, 4.0]}
df = pd.DataFrame(d).set_index('Index Title').rename_axis('Col Name', axis=1)
print (df)
Col Name     Column 1
Index Title          
Apples            1.0
Oranges           2.0
Puppies           3.0
Ducks             4.0

print (df.index.name)
Index Title

print (df.columns.name)
Col Name
print df.rename_axis('foo').rename_axis("bar", axis="columns")
bar      Column 1
foo              
Apples        1.0
Oranges       2.0
Puppies       3.0
Ducks         4.0

print df.rename_axis('foo').rename_axis("bar", axis=1)
bar      Column 1
foo              
Apples        1.0
Oranges       2.0
Puppies       3.0
Ducks         4.0

От версии pandas 0.24.0+ возможно использование параметра indexи columns:

df = df.rename_axis(index='foo', columns="bar")
print (df)
bar      Column 1
foo              
Apples        1.0
Oranges       2.0
Puppies       3.0
Ducks         4.0

Удаление имен индекса и столбцов означает установить его на None:

df = df.rename_axis(index=None, columns=None)
print (df)
         Column 1
Apples        1.0
Oranges       2.0
Puppies       3.0
Ducks         4.0

Если MultiIndexтолько в индексе:

mux = pd.MultiIndex.from_arrays([['Apples', 'Oranges', 'Puppies', 'Ducks'],
                                  list('abcd')], 
                                  names=['index name 1','index name 1'])


df = pd.DataFrame(np.random.randint(10, size=(4,6)), 
                  index=mux, 
                  columns=list('ABCDEF')).rename_axis('col name', axis=1)
print (df)
col name                   A  B  C  D  E  F
index name 1 index name 1                  
Apples       a             5  4  0  5  2  2
Oranges      b             5  8  2  5  9  9
Puppies      c             7  6  0  7  8  3
Ducks        d             6  5  0  1  6  0

print (df.index.name)
None

print (df.columns.name)
col name

print (df.index.names)
['index name 1', 'index name 1']

print (df.columns.names)
['col name']

df1 = df.rename_axis(('foo','bar'))
print (df1)
col name     A  B  C  D  E  F
foo     bar                  
Apples  a    5  4  0  5  2  2
Oranges b    5  8  2  5  9  9
Puppies c    7  6  0  7  8  3
Ducks   d    6  5  0  1  6  0

df2 = df.rename_axis('baz', axis=1)
print (df2)
baz                        A  B  C  D  E  F
index name 1 index name 1                  
Apples       a             5  4  0  5  2  2
Oranges      b             5  8  2  5  9  9
Puppies      c             7  6  0  7  8  3
Ducks        d             6  5  0  1  6  0

df2 = df.rename_axis(index=('foo','bar'), columns='baz')
print (df2)
baz          A  B  C  D  E  F
foo     bar                  
Apples  a    5  4  0  5  2  2
Oranges b    5  8  2  5  9  9
Puppies c    7  6  0  7  8  3
Ducks   d    6  5  0  1  6  0

Удаление имен индексов и столбцов означает установить его в None:

df2 = df.rename_axis(index=(None,None), columns=None)
print (df2)

           A  B  C  D  E  F
Apples  a  6  9  9  5  4  6
Oranges b  2  6  7  4  3  5
Puppies c  6  3  6  3  5  1
Ducks   d  4  9  1  3  0  5

В MultiIndexиндексах и столбцах необходимо работать с.names вместо .nameи набора по списку или кортежи:

mux1 = pd.MultiIndex.from_arrays([['Apples', 'Oranges', 'Puppies', 'Ducks'],
                                  list('abcd')], 
                                  names=['index name 1','index name 1'])


mux2 = pd.MultiIndex.from_product([list('ABC'),
                                  list('XY')], 
                                  names=['col name 1','col name 2'])

df = pd.DataFrame(np.random.randint(10, size=(4,6)), index=mux1, columns=mux2)
print (df)
col name 1                 A     B     C   
col name 2                 X  Y  X  Y  X  Y
index name 1 index name 1                  
Apples       a             2  9  4  7  0  3
Oranges      b             9  0  6  0  9  4
Puppies      c             2  4  6  1  4  4
Ducks        d             6  6  7  1  2  8

Множественное число необходимо для проверки / установки значений:

print (df.index.name)
None

print (df.columns.name)
None

print (df.index.names)
['index name 1', 'index name 1']

print (df.columns.names)
['col name 1', 'col name 2']

df1 = df.rename_axis(('foo','bar'))
print (df1)
col name 1   A     B     C   
col name 2   X  Y  X  Y  X  Y
foo     bar                  
Apples  a    2  9  4  7  0  3
Oranges b    9  0  6  0  9  4
Puppies c    2  4  6  1  4  4
Ducks   d    6  6  7  1  2  8

df2 = df.rename_axis(('baz','bak'), axis=1)
print (df2)
baz                        A     B     C   
bak                        X  Y  X  Y  X  Y
index name 1 index name 1                  
Apples       a             2  9  4  7  0  3
Oranges      b             9  0  6  0  9  4
Puppies      c             2  4  6  1  4  4
Ducks        d             6  6  7  1  2  8

df2 = df.rename_axis(index=('foo','bar'), columns=('baz','bak'))
print (df2)
baz          A     B     C   
bak          X  Y  X  Y  X  Y
foo     bar                  
Apples  a    2  9  4  7  0  3
Oranges b    9  0  6  0  9  4
Puppies c    2  4  6  1  4  4
Ducks   d    6  6  7  1  2  8

Удаление имен индексов и столбцов означает установить его в None:

df2 = df.rename_axis(index=(None,None), columns=(None,None))
print (df2)

           A     B     C   
           X  Y  X  Y  X  Y
Apples  a  2  0  2  5  2  0
Oranges b  1  7  5  5  4  8
Puppies c  2  4  6  3  6  5
Ducks   d  9  6  3  9  7  0

И решение @Jeff:

df.index.names = ['foo','bar']
df.columns.names = ['baz','bak']
print (df)

baz          A     B     C   
bak          X  Y  X  Y  X  Y
foo     bar                  
Apples  a    3  4  7  3  3  3
Oranges b    1  2  5  8  1  0
Puppies c    9  6  3  9  6  3
Ducks   d    3  2  1  0  1  0
jezrael
источник
7
Я считаю, что это должен быть принятый ответ для текущих версий панд
Томас Фаускангер
2
Также стоит отметить, как Фил предлагает в другом df.index.rename('foo', inplace=True)см. Pandas.pydata.org/pandas-docs/stable/generated/…
Томас Фаускангер,
Я импортировал словарь для использования в качестве кадра данных, поэтому столбец индекса был автоматически установлен на Нет, а строки пронумерованы 1-10. Но я хотел бы назначить столбец «имена» в качестве индекса. И если возможно, делайте это во время заговора. Возможно ли это сделать и возможно ли это сделать на лету, пока мы строим сюжет?
Токио тоже
28

df.index.name должен сделать свое дело.

В Python есть dirфункция, которая позволяет вам запрашивать атрибуты объекта. dir(df.index)было полезно здесь.

lazy1
источник
17

Используйте df.index.rename('foo', inplace=True)для установки имени индекса.

Кажется, этот API доступен начиная с версии 0.13 .

Фил
источник
1
Ух ты ... элегантное решение!
Alfonso
13

Если вы не хотите создавать новую строку, а просто поместите ее в пустую ячейку, используйте:

df.columns.name = 'foo'

В противном случае используйте:

df.index.name = 'foo'
Кит
источник
1
Просто нашел, что это имя для имен столбцов. Неудивительно, что установка df.index.name дает вам новый уровень. Спасибо!
Чарльз
6

df.columns.values также дайте нам имена столбцов

pvarma
источник
1

Решение для мультииндексов находится внутри циклопического ответа Джезраэль, но мне потребовалось некоторое время, чтобы найти его, поэтому я публикую новый ответ:

df.index.names дает имена мультииндекса (как Frozenlist).

Несчастный кот
источник
1

Чтобы просто получить имена столбцов индекса df.index.names будет работать как для одного индекса, так и для MultiIndex, начиная с самой последней версии панд.

Как человек, который нашел это, пытаясь найти лучший способ получить список имен индексов + имен столбцов, я нашел бы этот ответ полезным:

names = list(filter(None, df.index.names + df.columns.values.tolist()))

Это работает без индекса, индекса с одним столбцом или MultiIndex. Это позволяет избежать вызова метода reset_index (), который приводит к ненужному снижению производительности для такой простой операции. Я удивлен, что нет встроенного метода для этого (который я встречал). Я предполагаю, что мне это нужно чаще, потому что я перемещаю данные из баз данных, где индекс информационного кадра отображается на первичный / уникальный ключ, но на самом деле это просто еще один столбец для меня.

totalhack
источник
1

Задание имени индекса также может быть выполнено при создании:

pd.DataFrame(data={'age': [10,20,30], 'height': [100, 170, 175]}, index=pd.Series(['a', 'b', 'c'], name='Tag'))
dusiod
источник