Pandas: Selecting consecutive columns, rows
22. Block Selection of Rows and Columns
Write a Pandas program to select consecutive columns and also select rows with Index label 0 to 9 with some columns from world alcohol consumption dataset.
Test Data:
Year WHO region Country Beverage Types Display Value 0 1986 Western Pacific Viet Nam Wine 0.00 1 1986 Americas Uruguay Other 0.50 2 1985 Africa Cte d'Ivoire Wine 1.62 3 1986 Americas Colombia Beer 4.27 4 1987 Americas Saint Kitts and Nevis Beer 1.98
Sample Solution:
Python Code :
import pandas as pd
# World alcohol consumption data
w_a_con = pd.read_csv('world_alcohol.csv')
print("World alcohol consumption sample data:")
print(w_a_con.head())
print("\nSelect consecutive columns:")
print(w_a_con.loc[:,"Country":"Display Value"].head())
print("\nAlternate command:")
print(w_a_con.iloc[:,2:5].head())
print("\nSelect rows with Index label 0 to 9 with specific columns:")
print(w_a_con.loc[0:9,["Year","Country","Display Value"]])
Sample Output:
World alcohol consumption sample data:
Year WHO region ... Beverage Types Display Value
0 1986 Western Pacific ... Wine 0.00
1 1986 Americas ... Other 0.50
2 1985 Africa ... Wine 1.62
3 1986 Americas ... Beer 4.27
4 1987 Americas ... Beer 1.98
[5 rows x 5 columns]
Select consecutive columns:
Country Beverage Types Display Value
0 Viet Nam Wine 0.00
1 Uruguay Other 0.50
2 Cte d'Ivoire Wine 1.62
3 Colombia Beer 4.27
4 Saint Kitts and Nevis Beer 1.98
Alternate command:
Country Beverage Types Display Value
0 Viet Nam Wine 0.00
1 Uruguay Other 0.50
2 Cte d'Ivoire Wine 1.62
3 Colombia Beer 4.27
4 Saint Kitts and Nevis Beer 1.98
Select rows with Index label 0 to 9 with specific columns:
Year Country Display Value
0 1986 Viet Nam 0.00
1 1986 Uruguay 0.50
2 1985 Cte d'Ivoire 1.62
3 1986 Colombia 4.27
4 1987 Saint Kitts and Nevis 1.98
5 1987 Guatemala 0.00
6 1987 Mauritius 0.13
7 1985 Angola 0.39
8 1986 Antigua and Barbuda 1.55
9 1984 Nigeria 6.10
Click to download world_alcohol.csv
For more Practice: Solve these Related Problems:
- Write a Pandas program to slice a block of consecutive columns and rows (rows 0–9) and then reorder the columns.
- Write a Pandas program to extract a continuous block of rows and columns using loc, then verify the dimensions of the subset.
- Write a Pandas program to select a block of consecutive columns and the first 10 rows, and then rename the columns in the subset.
- Write a Pandas program to slice a DataFrame to obtain a specific rectangular area and then compute summary statistics for that block.
Go to:
Previous: Write a Pandas program to filter rows based on row numbers ended with 0, like 0, 10, 20, 30 from world alcohol consumption dataset.
Next: Write a Pandas program to rename all and only some of the column names from world alcohol consumption dataset.
Python Code Editor:
Have another way to solve this solution? Contribute your code (and comments) through Disqus.
What is the difficulty level of this exercise?
Test your Programming skills with w3resource's quiz.