Как вставить столбец в dataframe pandas

Часто вам может потребоваться вставить новый столбец в DataFrame pandas. К счастью, это легко сделать с помощью функции pandas Insert() , которая использует следующий синтаксис:

вставить (местоположение, столбец, значение, разрешить_дупликаты = False)

Золото:

• loc: индекс, в который нужно вставить столбец. Первый столбец равен 0.
• столбец: имя, которое будет присвоено новому столбцу.
• значение: Массив значений для нового столбца.
• разрешить_дупликаты: разрешить или запретить совпадение имени нового столбца с именем существующего столбца. По умолчанию установлено значение Ложь.

В этом уроке показано несколько примеров практического использования этой функции.

Пример 1. Вставка нового столбца в качестве первого столбца.

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

 import pandas as pd

#createDataFrame
df = pd.DataFrame({'points': [25, 12, 15, 14, 19],
                   'assists': [5, 7, 7, 9, 12],
                   'rebounds': [11, 8, 10, 6, 6]})

#view DataFrame
df
        points assists rebounds
0 25 5 11
1 12 7 8
2 15 7 10
3 14 9 6
4 19 12 6

#insert new column 'player' as first column
player_vals = ['A', 'B', 'C', 'D', 'E']
df. insert (loc= 0 , column='player', value=player_vals)
df

        player points assists rebounds
0 to 25 5 11
1 B 12 7 8
2 C 15 7 10
3 D 14 9 6
4 E 19 12 6

Пример 2. Вставка нового столбца в качестве среднего столбца.

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

 import pandas as pd

#createDataFrame
df = pd.DataFrame({'points': [25, 12, 15, 14, 19],
                   'assists': [5, 7, 7, 9, 12],
                   'rebounds': [11, 8, 10, 6, 6]})

#insert new column 'player' as third column
player_vals = ['A', 'B', 'C', 'D', 'E']
df. insert (loc= 2 , column='player', value=player_vals)
df

        points assists player rebounds
0 25 5 A 11
1 12 7 B 8
2 15 7 C 10
3 14 9 D 6
4 19 12 E 6

Пример 3. Вставка нового столбца в качестве последнего столбца.

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

 import pandas as pd

#createDataFrame
df = pd.DataFrame({'points': [25, 12, 15, 14, 19],
                   'assists': [5, 7, 7, 9, 12],
                   'rebounds': [11, 8, 10, 6, 6]})

#insert new column 'player' as last column
player_vals = ['A', 'B', 'C', 'D', 'E']
df. insert (loc= len (df.columns) , column='player', value=player_vals)
df

        points assists player rebounds
0 25 5 A 11
1 12 7 B 8
2 15 7 C 10
3 14 9 D 6
4 19 12 E 6

Обратите внимание, что использование len(df.columns) позволяет вам вставить новый столбец в качестве последнего столбца любого dataFrame, независимо от того, сколько столбцов он может содержать.

Полную документацию функции Insert() вы можете найти здесь .