Select multiple columns in Python Pandas
David Y.
The Problem
How do I select multiple columns from an existing DataFrame and create a new DataFrame with them?
The Solution
We can do this by creating a list of the column names we want and passing them to the DataFrame constructor method, along with the original DataFrame. The code below shows an example:
import pandas # Our main DataFrame main = pandas.DataFrame([["apple", 1, 2], ["orange", 3, 4], ["pear", 5, 6]], columns=["product", "cost_price", "sale_price"]) print(main) print("\n") # Our smaller DataFrame subset = pandas.DataFrame(main, columns=["product", "sale_price"]) print(subset)
This code will produce the following output:
product cost_price sale_price 0 apple 1 2 1 orange 3 4 2 pear 5 6 product sale_price 0 apple 2 1 orange 4 2 pear 6
Ask a questionImprove this Answer
Related Answers
- Access the index of a for loop
- Accessing the Index in a `for` Loop in Python
- Append vs. extend in Python
- Change the size of figures drawn with Matplotlib in Python
- Check if a dictionary contains a key in Python
- Check if a list is empty in Python
- Concatenate two lists in Python
- Convert a list to a string in Python
- Convert an integer to a string in Python
- Convert bytes to a string in Python
- Convert a string containing a date into datetime in Python
- Delete a file or folder in Python
- Determine the type of a Python variable
- Difference between `@staticmethod` and `@classmethod` function decorators in Python
- Extract a substring from a string in Python
- How to manage feature flags without making performance worse
- Find the current directory and executed file's directory in Python
- Get the length of a list in Python
- How do I parse a string to a float or int?
- What does `if __name__ == "__main__":` do?
- Iterate over rows in a Pandas DataFrame in Python
- Limit floats to two decimal points in Python
- Is There a List of pytz Timezones?
- Lowercase a string in Python
- Making a list out of a list of lists in Python?
- Pandas DataFrame row count in Python
- Add new keys to a dictionary in Python
- Sort a dictionary by value in Python
- Iterate over a dictionary in Python
- List files in a directory using Python
- How Do I Merge Two Dictionaries in a Single Expression (Take Union of Dictionaries)?
- How Can I Safely Create a Nested Directory?
- Python slice notation
- Does Python Have a Ternary Conditional Operator?
- What Does the `yield` Keyword in Python Do?
- Read a file line by line into a list Python
- Remove element from a list by index in Python
- Rename columns in a Python Pandas DataFrame
- Running External Programs in Python
- Select rows from a Python Pandas DataFrame based on column values
- TypeError 'str' Object Does Not Support Item Assignment
- Time delay in Python
- UnboundLocalError Local Variable 'index' Referenced Before Assignment
- Upgrade specific `pip` packages in Python
- Use global variables in Python functions
- What are Python Metaclasses?