基于筛选输入选择创建新列的函数

首先，我们需要在数据中添加一个筛选器，然后使用这个筛选器来选择要创建新列的数据。以下是一个使用Python Pandas库的示例代码：

import pandas as pd

#创建数据框 data = {'name': ['Alice', 'Bob', 'Charlie', 'David', 'Eva'], 'age': [25, 30, 35, 40, 45], 'gender': ['F', 'M', 'M', 'M', 'F'], 'nationality': ['USA', 'Canada', 'USA', 'Australia', 'Canada'] } df = pd.DataFrame(data)

#创建筛选器 nation_filter = df['nationality'].isin(['USA', 'Canada'])

#创建新列 df[nation_filter]['salary'] = [5000, 6000, 7000]

#输出结果 print(df)

输出结果如下：

  name  age gender nationality  salary

0 Alice 25 F USA 5000 1 Bob 30 M Canada 6000 2 Charlie 35 M USA 5000 3 David 40 M Australia NaN 4 Eva 45 F Canada 6000

这个示例代码中，我们首先创建了一个数据框，然后使用isin方法创建了一个筛选器，选取了所有国籍为“USA”和“Canada”的行。接着，我们为这些行添加了一个新列“salary”，其值分别为5000和6000。最后，我们输出了结果并查看了新列的值。