Select specific columns in Pandas DataFrame

Pandas: Advanced Indexing and Slicing Exercise-2 with Solution

Select Specific Columns:

Write a Pandas program to select only the 'X' and 'Y' columns from the DataFrame.

Sample Solution :

Python Code :

import pandas as pd

# Create a DataFrame
df = pd.DataFrame({
    'X': [1, 6, 8, 3, 7],
    'Y': [5, 2, 9, 4, 1],
    'Z': [3, 8, 6, 2, 9]
})

# Select 'X' and 'Y' columns
result = df[['X', 'Y']]
print(result)

Output:

   X  Y
0  1  5
1  6  2
2  8  9
3  3  4
4  7  1

Explanation:

• Import pandas library.
• Create a DataFrame.
• Select columns 'X' and 'Y'.
• Print the results.

Python-Pandas Code Editor:

Have another way to solve this solution? Contribute your code (and comments) through Disqus.

Previous: Select rows with specific condition in Pandas DataFrame.
Next: Set MultiIndex and access data in Pandas DataFrame/