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.
Python Code Editor:
Have another way to solve this solution? Contribute your code (and comments) through Disqus.
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.
What is the difficulty level of this exercise?
Test your Programming skills with w3resource's quiz.