Pandas Adding - Deleting columns and Index Operations

Pandas Adding - Deleting columns and Index Operations. The following shows how to add column, delete column from pandas DataFrame and various index Operations using pandas module.

In [1]:

import pandas as pd

In [2]:

data_list = [0,1,2,3,4,5,6,7,8,9]

data_frame = pd.DataFrame(data_list)
data_frame

Out[2]:

Change the name of the column:

In [3]:

data_frame.columns = ['rev']
data_frame

Out[3]:

Add a column:

In [4]:

data_frame['new_col'] = 5
data_frame

Out[4]:

Modify our new column:

In [5]:

data_frame['new_col'] = data_frame['new_col'] + 1
data_frame

Out[5]:

In [6]:

del data_frame['new_col']
data_frame

Out[6]:

Adding couple of columns:

In [7]:

data_frame['test_col'] = 3
data_frame['another_col'] = data_frame['rev']
data_frame

Out[7]:

Change the name of the index:

In [8]:

index = ['a','b','c','d','e','f','g','h','i','j']
data_frame.index = index
data_frame

Out[8]:

Select pieces of the dataframe using loc:

In [9]:

data_frame.loc['a']

Out[9]:

rev            0
test_col       3
another_col    0
Name: a, dtype: int64

In [10]:

data_frame.loc['a':'d']

Out[10]:

In [11]:

data_frame.iloc[0:4]

Out[11]:

Select using the column name:

In [12]:

data_frame['rev']

Out[12]:

a    0
b    1
c    2
d    3
e    4
f    5
g    6
h    7
i    8
j    9
Name: rev, dtype: int64

In [13]:

data_frame[['rev', 'test_col']]

Out[13]:

In [14]:

data_frame.loc[data_frame.index[0:3], 'rev']

Out[14]:

a    0
b    1
c    2
Name: rev, dtype: int64

In [15]:

data_frame.loc[data_frame.index[0:3], ['rev', 'test_col']]

Out[15]:

In [16]:

data_frame.loc[data_frame.index[5:],'another_col']

Out[16]:

f    5
g    6
h    7
i    8
j    9
Name: another_col, dtype: int64

In [17]:

data_frame.loc[data_frame.index[:3],['another_col', 'test_col']]

Out[17]:

Select top N number of records:

In [18]:

n = 5
data_frame.head(n)

Out[18]:

Select bottom N number of records:

In [19]:

n = 5
data_frame.tail(5)

Out[19]: