Як вставити стовпець у pandas dataframe

Часто ви можете вставити новий стовпець у pandas DataFrame. На щастя, це легко зробити за допомогою функції pandas insert() , яка використовує такий синтаксис:

вставити(лок, стовпець, значення, дозволити_дуплікати=Хибне)

золото:

• loc: індекс, у який потрібно вставити стовпець. Перший стовпець – 0.
• стовпець: назва для нового стовпця.
• значення: масив значень для нового стовпця.
• allow_duplicates: дозволяти чи забороняти новій назві стовпця відповідати існуючій назві стовпця. За замовчуванням значення False.

Цей підручник демонструє кілька прикладів практичного використання цієї функції.

Приклад 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() тут .