dodawanie i usuwanie kolumny dataframe

W pandas istnieje kilka sposobów na zmianę kształtu danych, takich jak transponowanie dataframe’u, wstawianie lub usuwanie wierszy lub kolumn, czy też łączenie kilku dataframe’ów w jeden.

Jednym z najprostszych sposobów zmiany kształtu danych jest transponowanie dataframe’u za pomocą metody transpose(). Transponowanie dataframe’u polega na zamianie wierszy z kolumnami i odwrotnie.

Przykład:

import pandas as pd

# Tworzenie prostego dataframe'u
df = pd.DataFrame({'col1': [1, 2, 3], 'col2': [4, 5, 6], 'col3': [7, 8, 9]})
print(df)

   col1  col2  col3
0     1     4     7
1     2     5     8
2     3     6     9

# Transponowanie dataframe'u
df_t = df.transpose()
print(df_t)

      0  1  2
col1  1  2  3
col2  4  5  6
col3  7  8  9

Przykład dodawania i usuwania kolumny:

import pandas as pd


df = pd.DataFrame({'col1': [1, 2, 3], 'col2': [4, 5, 6], 'col3': [7, 8, 9]})
print(df)


   col1  col2  col3
0     1     4     7
1     2     5     8
2     3     6     9

# Wstawienie nowej kolumny o etykiecie 'col4' i wartościach [10, 11, 12] na pozycji 2
df.insert(2, 'col4', [10, 11, 12], True)
print(df)


   col1  col2  col4  col3
0     1     4    10     7
1     2     5    11     8
2     3     6    12     9

# Usunięcie kolumny o etykiecie 'col4'
df.drop('col4', axis=1, inplace=True)
print(df)

   col1  col2  col3
0     1     4     7
1     2     5     8
2     3     6     9