Add a new column to a DataFrame in Python Pandas
David Y.
The Problem
How do I add a new column to an existing Pandas DataFrame? The new column should have the same number of rows as existing columns.
The Solution
We can do this using the DataFrame.assign method. When called on a given DataFrame, this method creates a copy of that DataFrame with one or more additional columns added.
The new columns can contain manually specified values, or values can be computed for each row based on other columns. Look at the following example code:
import pandas # Create a dataframe with products and their cost prices products = pandas.DataFrame([["apple", 2.0], ["orange", 3.0], ["pear", 4.0]], columns=["product", "cost_price"]) print(products) # Add a stock count column with manual data products = products.assign(stock_count=[50, 40, 30]) print(products) # Add a sale price column that calculates a 50% markup products = products.assign(sale_price=lambda row: row.cost_price * 1.5) print(products)
This code will print the products DataFrame with two, three and then four columns. Note the use of a lambda expression in the creation of the sale_price column – the values for this column are created by running the function on each row.
Ask a questionImprove this Answer
Related Answers
- Check if a string is empty in Python
- Access environment variables in Python
- 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 for NaN values in Python
- Check if a dictionary contains a key in Python
- Check if a list is empty in Python
- Concatenate two lists in Python
- How Can I Safely Create a Nested Directory?
- Python slice notation
- Does Python Have a Ternary Conditional Operator?
- Raise an exception in Python
- How Do I Merge Two Dictionaries in a Single Expression (Take Union of Dictionaries)?
- 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
- Remove whitespace from a string in Python
- Rename columns in a Python Pandas DataFrame
- Convert a list to a string in Python
- Convert a string to bytes 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
- Copy files in Python
- Count occurrences of item in Python list
- Declare an array in Python
- Define a two-dimensional array in Python
- Delete a column from a DataFrame in Python Pandas
- Delete a file or folder in Python
- Delete element from a dictionary 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
- Fastest way to check if a list contains an item in Python
- How to manage feature flags without making performance worse
- Find all files in a directory with a given extension using Python
- Find item in list in Python
- Find the current directory and executed file's directory in Python
- Generate random integers within a range in Python
- Get key by value in dictionary Python
- Get the current script file's full directory path in Python
- Get the current time in Python
- Get the last element of a list in Python
- Get the length of a list in Python
- Get unique values from a list in Python
- How do I append one string to another in Python?
- How do I parse a string to a float or int?
- How do I remove a trailing newline from a string in Python
- What does `if __name__ == "__main__":` do?
- Import files from a different folder in Python
- Indentation errors in Python
- Install a specific version of a Python package using PIP
- Install Python packages from a local package index
- Invalid literal for int error in Python
- Iterate over rows in a Pandas DataFrame in Python
- Limit floats to two decimal points in Python
- Is There a List of pytz Timezones?
- Logical AND (&&) in Python
- Lowercase a string in Python
- Making a list out of a list of lists in Python?
- Measure elapsed time in Python
- Pandas DataFrame row count in Python
- Print multiple times without newlines or spaces in Python
- Add new keys to a dictionary in Python
- Sort a dictionary by value in Python
- Python equivalent of a ternary operator: if-then-else in one line
- Iterate over a dictionary in Python
- List files in a directory using Python
- Running External Programs in Python
- Save a Matplotlib plot to a file in Python
- Select multiple columns in Python Pandas
- Select rows from a Python Pandas DataFrame based on column values
- Slicing in Python
- Split statements across multiple lines in Python code
- TypeError 'str' Object Does Not Support Item Assignment
- `super()` and `__init__()` in Python
- Switch-case statement equivalents in Python
- 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?
- Write one or more lines to a file in Python