Extract One & Multiple pandas DataFrame Columns by Index in Python (2 Examples)
This tutorial shows how to get one or more variables by index in Python programming.
Preparing the Examples
import pandas as pd # Load pandas library |
import pandas as pd # Load pandas library
my_df = pd.DataFrame({'A':range(2, 8), # Construct pandas DataFrame 'B':range(22, 16, - 1), 'C':['q', 'w', 'e', 'r', 't', 'z'], 'D':range(10, 16), 'E':['f', 'f', 'g', 'g', 'f', 'f'], 'F':range(19, 25)}) print(my_df) # A B C D E F # 0 2 22 q 10 f 19 # 1 3 21 w 11 f 20 # 2 4 20 e 12 g 21 # 3 5 19 r 13 g 22 # 4 6 18 t 14 f 23 # 5 7 17 z 15 f 24 |
my_df = pd.DataFrame({'A':range(2, 8), # Construct pandas DataFrame 'B':range(22, 16, - 1), 'C':['q', 'w', 'e', 'r', 't', 'z'], 'D':range(10, 16), 'E':['f', 'f', 'g', 'g', 'f', 'f'], 'F':range(19, 25)}) print(my_df) # A B C D E F # 0 2 22 q 10 f 19 # 1 3 21 w 11 f 20 # 2 4 20 e 12 g 21 # 3 5 19 r 13 g 22 # 4 6 18 t 14 f 23 # 5 7 17 z 15 f 24
Example 1: Selecting a Single pandas DataFrame Column by Index
df_one = my_df.iloc[:, [3]] # Extract one column print(df_one) # D # 0 10 # 1 11 # 2 12 # 3 13 # 4 14 # 5 15 |
df_one = my_df.iloc[:, [3]] # Extract one column print(df_one) # D # 0 10 # 1 11 # 2 12 # 3 13 # 4 14 # 5 15
Example 2: Selecting Multiple pandas DataFrame Columns by Index
df_multiple = my_df.iloc[:, [0, 1, 3, 5]] # Extract multiple columns print(df_multiple) # A B D F # 0 2 22 10 19 # 1 3 21 11 20 # 2 4 20 12 21 # 3 5 19 13 22 # 4 6 18 14 23 # 5 7 17 15 24 |
df_multiple = my_df.iloc[:, [0, 1, 3, 5]] # Extract multiple columns print(df_multiple) # A B D F # 0 2 22 10 19 # 1 3 21 11 20 # 2 4 20 12 21 # 3 5 19 13 22 # 4 6 18 14 23 # 5 7 17 15 24
Related Tutorials
Have a look at the following Python tutorials. They focus on topics such as indices and data inspection: