Pandas: create two new columns in a dataframe with values calculated from a pre-existing column

by

I’d just use zip:

In [1]: from pandas import *

In [2]: def calculate(x):
   ...:     return x*2, x*3
   ...: 

In [3]: df = DataFrame({'a': [1,2,3], 'b': [2,3,4]})

In [4]: df
Out[4]: 
   a  b
0  1  2
1  2  3
2  3  4

In [5]: df["A1"], df["A2"] = zip(*df["a"].map(calculate))

In [6]: df
Out[6]: 
   a  b  A1  A2
0  1  2   2   3
1  2  3   4   6
2  3  4   6   9

More Related Contents:

  1. How to deal with SettingWithCopyWarning in Pandas
  2. Apply multiple functions to multiple groupby columns
  3. Scatter plots in Pandas/Pyplot: How to plot by category [duplicate]
  4. Using Pandas to pd.read_excel() for multiple worksheets of the same workbook
  5. How to add pandas data to an existing csv file?
  6. Pandas timeseries plot setting x-axis major and minor ticks and labels
  7. How do I operate on a DataFrame with a Series for every column?
  8. Improve subplot size/spacing with many subplots in matplotlib
  9. Pandas split column into multiple columns by comma
  10. Repeat rows in a pandas DataFrame based on column value
  11. Accessing every 1st element of Pandas DataFrame column containing lists
  12. How to use groupby to concatenate strings in python pandas?
  13. Add months to a date in Pandas
  14. Horizontal stacked bar plot and add labels to each section
  15. How to flatten a nested JSON recursively, with flatten_json
  16. Python – How to convert JSON File to Dataframe
  17. How to remove timezone from a Timestamp column in a pandas dataframe
  18. Pandas lookup from one of multiple columns, based on value
  19. How to add a suffix (or prefix) to each column name?
  20. dropping rows from dataframe based on a “not in” condition [duplicate]
  21. How to delete multiple pandas (python) dataframes from memory to save RAM?
  22. Adding two Series with NaNs
  23. Create Column with ELIF in Pandas
  24. ‘utf-8’ codec can’t decode byte 0x92 in position 18: invalid start byte
  25. Pandas – Compute z-score for all columns
  26. Find date range overlap in python
  27. How do I delete a column that contains only zeros in Pandas?
  28. Updating a value in a pandas dataframe in an iterrows loop
  29. How to convert DataFrame.append() to pandas.concat()?
  30. How to efficiently assign unique ID to individuals with multiple entries based on name in very large df

Leave a Comment