Question:
Creating a list in a dataframe column from two other columns' values

Summary:

You can create a list of a range of numbers as a column in a DataFrame using the popular Python library called Pandas. 

In this example:


  • start_number: the starting number of the range.

  • end_number: the ending number of the range.

  • step: the step size between numbers in the range.

The range function is used to generate a sequence of numbers, and then you can pass this sequence as a column to the DataFrame constructor.


Replace 'Column_Name' with the desired name for your column. Adjust the start_number, end_number, and step variables according to your requirements.


Solution:

Sample code:

import pandas as pd


# Define the range of numbers

start_number = 1

end_number = 10

step = 1


# Create a DataFrame with a column of the range of numbers

df = pd.DataFrame({'Column_Name': range(start_number, end_number + 1, step)})


# Display the DataFrame

print(df)


Working code after DataFrame.apply on rows.

# create 1 for long, else -1

d = np.where(df['dir'].eq('long'), 1, -1)[:,None]


# convert a and b to numpy as a column vector

a = df['a'].to_numpy()[:,None]

b = df['b'].to_numpy()[:,None]


# combine

N = 3

x = np.arange(1, N+1)

df['x'] = list(a + d*b*x)


Output:

    a  b    dir             x

0   20  2   long  [22, 24, 26]

1  100  3  short  [97, 94, 91]


Credit:> Stack Overflow


Suggested blogs:

>How .transform handle the splitted groups?

>Can I use VS Code's launch config to run a specific python file?

>Python: How to implement plain text to HTML converter?

>How to write specific dictionary list for each cycle of loop?

>Reading a shapefile from Azure Blob Storage and Azure Databricks

>How to replace a value by another in certain columns of a 3d Numpy array?

>How to fix "segmentation fault" issue when using PyOpenGL on Mac?


Submit
0 Answers
Top Questions
26-02-2024

How to solve maximum sum problem using Python3

06-01-2024

How to Install mariaDB client efficiently inside docker?

23-01-2024

Return an array of leaders in order of their appearance using Python3

19-12-2023

How remove residual lines after merge two regions using Geopandas in python?

15-01-2024

Creating a list in a dataframe column from two other columns' values

Related Questions

Can I call an API to find out the random seed in Python `hash()` function?
img
How to allign two plots in Matplotlib?
img
How to solve maximum sum problem using Python3
img
How to Reverse a sublist of a linked list in Python?
img
Creating a pivot table by 6 month interval rather than year
img

What Our Clients Say About Us

Client satisfaction is our ultimate goal. Here are some kind words of our precious clients they have used to express their satisfaction with our service.

Why Choose Us ?

We endow businesses with flexible engagement models based on their unique needs. Our strength lies in state-of-the-art technology and affordable consulting services. Try us for fast POCs, full-fledged applications, or technology consulting. Always available for your service.